def test_store_sqlite_write_d(self) -> None:
f1 = Frame.from_dict(dict(a=(1, 2, 3), b=(4, 5, 6)),
index=('x', 'y', 'z'),
name='f2')
frames = (f1, )
with temp_file('.sqlite') as fp:
config = StoreConfig(include_index=False)
st1 = StoreSQLite(fp)
st1.write(((f.name, f) for f in frames), config=config)
f2 = st1.read(f1.name, config=config)
self.assertEqual(f2.to_pairs(0), (('a', ((0, 1), (1, 2), (2, 3))),
('b', ((0, 4), (1, 5), (2, 6)))))
# getting the default config
f3 = st1.read(f1.name, config=None)
self.assertEqual(f3.to_pairs(0), (('a', ((0, 1), (1, 2), (2, 3))),
('b', ((0, 4), (1, 5), (2, 6)))))
def test_store_sqlite_write_b(self) -> None:
f1 = Frame.from_dict(
dict(
x=(Fraction(3,2), Fraction(1,2), Fraction(2,3), Fraction(3,7)),
y=(3,4,-5,-3000)),
index=IndexHierarchy.from_product(('I', 'II'), ('a', 'b')),
name='f1-dash')
frames = (f1,)
with temp_file('.sqlite') as fp:
st1 = StoreSQLite(fp)
st1.write((f.name, f) for f in frames)
config = StoreConfig.from_frame(f1)
f_loaded = st1.read(f1.name, config=config)
# for now, Fractions come back as strings
self.assertEqual(
f_loaded['x'].to_pairs(),
((('I', 'a'), '3/2'), (('I', 'b'), '1/2'), (('II', 'a'), '2/3'), (('II', 'b'), '3/7'))
)
def test_store_sqlite_write_a(self) -> None:
f1 = Frame.from_dict(dict(x=(None, -np.inf, np.inf, None),
y=(3, 4, -5, -3000)),
index=IndexHierarchy.from_product(('I', 'II'),
('a', 'b')),
name='f1')
f2 = Frame.from_dict(dict(a=(1, 2, 3), b=(4, 5, 6)),
index=('x', 'y', 'z'),
name='f2')
f3 = Frame.from_records(((10.4, 20.1, 50, 60), (50.1, 60.4, -50, -60)),
index=('p', 'q'),
columns=IndexHierarchy.from_product(
('I', 'II'), ('a', 'b')),
name='f3')
f4 = Frame.from_records(
(
(10, 20, 50, False, 10, 20, 50, False),
(50.0, 60.4, -50, True, 50.0, 60.4, -50, True),
(234, 44452, 0, False, 234, 44452, 0, False),
(4, -4, 2000, True, 4, -4, 2000, True),
(10, 20, 50, False, 10, 20, 50, False),
(50.0, 60.4, -50, True, 50.0, 60.4, -50, True),
(234, 44452, 0, False, 234, 44452, 0, False),
(4, -4, 2000, True, 4, -4, 2000, True),
),
index=IndexHierarchy.from_product(
('top', 'bottom'), ('far', 'near'), ('left', 'right')),
columns=IndexHierarchy.from_product(('I', 'II'), ('a', 'b'),
(1, 2)),
name='f4')
frames = (f1, f2, f3, f4)
with temp_file('.sqlite') as fp:
st1 = StoreSQLite(fp)
st1.write((f.name, f) for f in frames)
sheet_names = tuple(
st1.labels()) # this will read from file, not in memory
self.assertEqual(tuple(f.name for f in frames), sheet_names)
for i, name in enumerate(sheet_names):
f_src = frames[i]
config = StoreConfig.from_frame(f_src)
f_loaded = st1.read(name, config=config)
self.assertEqualFrames(f_src, f_loaded)
def test_store_sqlite_write_c(self) -> None:
f1 = Frame.from_dict(
dict(
x=np.array([1.2, 4.5, 3.2, 6.5], dtype=np.float16),
y=(3,4,-5,-3000)),
index=IndexHierarchy.from_product(('I', 'II'), ('a', 'b')),
name='f1')
frames = (f1,)
with temp_file('.sqlite') as fp:
st1 = StoreSQLite(fp)
st1.write((f.name, f) for f in frames)
config = StoreConfig.from_frame(f1)
f_loaded = st1.read(f1.name, config=config)
self.assertAlmostEqualItems(f_loaded['x'].to_pairs(),
((('I', 'a'), 1.2001953125), (('I', 'b'), 4.5), (('II', 'a'), 3.19921875), (('II', 'b'), 6.5))
)
