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(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)