def test_read_title_tsv(self): i = 0 with TsvReader('tsvimporter/tests/data/test.title.basics.tsv') as tsv: self.assertListEqual(tsv.columns, TITLE_DATA['columns']) for row in tsv.readline(): self.assertListEqual(row, TITLE_DATA['rows'][i]) i += 1
def test_read_name_tsv_in_gzip(self): i = 0 with TsvReader('tsvimporter/tests/data/test.name.basics.tsv.gz') as tsv: self.assertListEqual(tsv.columns, NAME_DATA['columns']) for row in tsv.readline(): self.assertListEqual(row, NAME_DATA['rows'][i]) i += 1
def handle(self, *args, **options): for path in options['path']: with TsvReader(path) as tsv: if 'primaryTitle' in tsv.columns: serializer = TitlesSerializer if 'primaryName' in tsv.columns: serializer = NamesSerializer self.save_to_db(tsv, serializer)
def test_open_wrong_extension(self): with self.assertRaises(ValueError): with TsvReader('s') as tsv: print(tsv)
def test_opening_non_existing_file_rises_error(self): with self.assertRaises(FileNotFoundError): with TsvReader('non-existing-file.tsv.gz') as tsv: print(tsv)