def test_fromtsv(): data = [b'foo\tbar', b'a\t1', b'b\t2', b'c\t2'] f = NamedTemporaryFile(mode='wb', delete=False) f.write(b'\n'.join(data)) f.close() expect = (('foo', 'bar'), ('a', '1'), ('b', '2'), ('c', '2')) actual = fromtsv(f.name, encoding='ascii') ieq(expect, actual) ieq(expect, actual) # verify can iterate twice
def test_zipsource(): # setup table = [('foo', 'bar'), ('a', '1'), ('b', '2')] totsv(table, 'tmp/issue_241.tsv') z = zipfile.ZipFile('tmp/issue_241.zip', mode='w') z.write('tmp/issue_241.tsv', 'data.tsv') z.close() # test actual = fromtsv(ZipSource('tmp/issue_241.zip', 'data.tsv')) ieq(table, actual)
def test_zipsource(): # setup table = [("foo", "bar"), ("a", "1"), ("b", "2")] totsv(table, "tmp/issue_241.tsv") z = zipfile.ZipFile("tmp/issue_241.zip", mode="w") z.write("tmp/issue_241.tsv", "data.tsv") z.close() # test actual = fromtsv(ZipSource("tmp/issue_241.zip", "data.tsv")) ieq(table, actual)
def test_fromtsv(): f = NamedTemporaryFile(delete=False) writer = csv.writer(f, delimiter='\t') table = (('foo', 'bar'), ('a', 1), ('b', 2), ('c', 2)) for row in table: writer.writerow(row) f.close() actual = fromtsv(f.name) expect = (('foo', 'bar'), ('a', '1'), ('b', '2'), ('c', '2')) ieq(expect, actual) ieq(expect, actual) # verify can iterate twice