def test_bus_init_a(self) -> None:
f1 = Frame.from_dict(
dict(a=(1,2), b=(3,4)),
index=('x', 'y'),
name='foo')
f2 = Frame.from_dict(
dict(a=(1,2,3), b=(4,5,6)),
index=('x', 'y', 'z'),
name='bar')
b1 = Bus.from_frames((f1, f2))
self.assertEqual(b1.keys().values.tolist(),
['foo', 'bar'])
with temp_file('.zip') as fp:
b1.to_zip_tsv(fp)
b2 = Bus.from_zip_tsv(fp)
f3 = b2['bar']
f4 = b2['foo']
# import ipdb; ipdb.set_trace()
zs = StoreZipTSV(fp)
zs.write(b1.items())
f3 = zs.read('foo')
self.assertEqual(
f3.to_pairs(0),
(('a', (('x', 1), ('y', 2))), ('b', (('x', 3), ('y', 4))))
)
def test_store_zip_tsv_a(self) -> None:
f1 = Frame.from_dict(dict(a=(1, 2), b=(3, 4)),
index=('x', 'y'),
name='foo')
f2 = Frame.from_dict(dict(a=(1, 2, 3), b=(4, 5, 6)),
index=('x', 'y', 'z'),
name='bar')
f3 = Frame.from_dict(dict(a=(10, 20), b=(50, 60)),
index=('p', 'q'),
name='baz')
with temp_file('.zip') as fp:
st = StoreZipTSV(fp)
st.write((f.name, f) for f in (f1, f2, f3))
labels = tuple(st.labels(strip_ext=False))
self.assertEqual(labels, ('foo.txt', 'bar.txt', 'baz.txt'))
for label, frame in ((f.name, f) for f in (f1, f2, f3)):
frame_stored = st.read(label)
self.assertEqual(frame_stored.shape, frame.shape)
self.assertTrue((frame_stored == frame).all().all())
self.assertEqual(frame.to_pairs(0), frame_stored.to_pairs(0))
def from_zip_tsv(cls, fp: PathSpecifier) -> 'Bus':
store = StoreZipTSV(fp)
return cls(cls._deferred_series(store.labels()), store=store)
def to_zip_tsv(self, fp: PathSpecifier) -> None:
store = StoreZipTSV(fp)
store.write(self.items())
def test_store_init_a(self) -> None:
with self.assertRaises(ErrorInitStore):
StoreZipTSV('test.txt') # must be a zip