def test_tubreader():
with tempfile.TemporaryDirectory() as tempfolder:
path = os.path.join(tempfolder, 'new')
inputs = ['name', 'age', 'pic']
types = ['str', 'float', 'str']
writer = TubWriter(path, inputs=inputs, types=types)
writer.run('will', 323, 'asdfasdf')
assert writer.get_num_records() == 1
reader = TubReader(path)
assert reader.get_num_records() == 1
record = reader.run('name', 'age', 'pic')
assert set(record) == set(['will', 323, 'asdfasdf'])
def test_tub_path(self):
tub = TubWriter(self.path, inputs=self.inputs, types=self.types)
print(tub.types, tub.inputs)
tub.run('will', 323, 'asdfasdf')