def test_insert(transformed, tmpdir):
t = transformed
expected = execute(t)
tpath = tmpdir / 'new_dir'
tpath.mkdir()
path = tpath / 'foo.h5'
assert not path.exists()
t = transformed[['time', 'ticker', 'avg']]
c = ibis.hdf5.connect(tpath)
c.insert('foo.h5', 'avg', t)
execute(t)
assert path.exists()
# readback
c = HDFClient(str(tpath)).database()
result = c.list_databases()
assert result == ['foo']
result = c.foo.avg.execute()
tm.assert_frame_equal(result, expected)
path = tpath / 'foo.h5'
assert path.exists()
def test_insert(transformed, tmpdir):
t = transformed
expected = execute(t)
tpath = tmpdir / 'new_dir'
tpath.mkdir()
path = tpath / 'foo.h5'
assert not path.exists()
t = transformed[['time', 'ticker', 'avg']]
c = ibis.hdf5.connect(tpath)
c.insert('foo.h5', 'avg', t)
execute(t)
assert path.exists()
# readback
c = HDFClient(str(tpath)).database()
result = c.list_databases()
assert result == ['foo']
result = c.foo.avg.execute()
tm.assert_frame_equal(result, expected)
path = tpath / 'foo.h5'
assert path.exists()
def test_read_with_projection(csv2, data):
t = csv2.csv_dir2.df
result = execute(t)
assert 'close' in result.columns
assert 'open' in result.columns
t = t[['time', 'ticker', 'close']]
result = execute(t)
assert 'close' in result.columns
assert 'open' not in result.columns
def test_read(parquet, data):
closes = parquet.pq.close
assert str(closes) is not None
result = closes.execute()
expected = data['close']
tm.assert_frame_equal(result, expected)
result = execute(closes)
tm.assert_frame_equal(result, expected)
def test_read(hdf, data):
closes = hdf.hdf_dir.prices.close
assert str(closes) is not None
result = closes.execute()
expected = data['close']
tm.assert_frame_equal(result, expected)
result = execute(closes)
tm.assert_frame_equal(result, expected)
def test_read(hdf, data):
closes = hdf.hdf_dir.prices.close
assert str(closes) is not None
result = closes.execute()
expected = data['close']
tm.assert_frame_equal(result, expected)
result = execute(closes)
tm.assert_frame_equal(result, expected)
def test_read(parquet, data):
closes = parquet.pq.close
assert str(closes) is not None
result = closes.execute()
expected = data['close']
tm.assert_frame_equal(result, expected)
result = execute(closes)
tm.assert_frame_equal(result, expected)
def test_read(csv, data):
closes = csv.csv_dir.close
assert str(closes) is not None
result = closes.execute()
expected = data['close']
# csv's don't preserve dtypes
expected['time'] = expected['time'].astype(str)
tm.assert_frame_equal(result, expected)
result = execute(closes)
tm.assert_frame_equal(result, expected)
def test_insert(transformed, tmpdir):
t = transformed
# csv's don't preserve dtypes
expected = execute(t)
expected['time'] = expected['time'].astype(str)
tpath = tmpdir / 'new_csv'
tpath.mkdir()
path = tpath / 'foo.csv'
assert not path.exists()
c = ibis.csv.connect(tpath)
c.insert('foo.csv', t)
assert path.exists()
# readback
t = CSVClient(str(tpath)).database()
result = t.list_tables()
assert result == ['foo']
result = t.foo.execute()
tm.assert_frame_equal(result, expected)
