def test_if_created(self): lyrics = self._create() expected = [lyrics] result = Lyrics.select() self.assertEqual(result, expected)
def _create(self, commit=True, **kwargs): data = { 'artist': 'Testudo', 'artist_url': 'http', 'music': 'Test', 'music_url': 'http://', 'text': 'Testing', 'translate': 'Testando', 'artist_tag': 'testudo', 'music_tag': 'test', 'created_date': datetime.now().date(), } data.update(kwargs) lyrics = Lyrics(**data) if commit: lyrics.save() return lyrics
def __clear_table(self): Lyrics.delete().execute()