def test_csv_node(): basepath = join(dirname(dirname(__file__)), 'example') client = ArrayClient(basepath) node = client.get_node('/csvs/sample') data = node.get() #better check later assert data.shape == (73,2)
def __init__(self): EstuarialDir = pjoin(expanduser('~'), '.estuarial') self.basedir = pjoin(os.path.dirname(__file__), '..', 'data', 'catalog', 'SQL_DATA') self.aclient = ArrayClient(basepath=self.basedir, localdatapath=EstuarialDir) self.aclient.set_logging(log)
def test_csv_node(): basepath = join(dirname(dirname(__file__)), 'example') client = ArrayClient(basepath) node = client.get_node('/csvs/sample') data = node.get() #better check later assert data.shape == (73, 2)
def test_hdf_node(): basepath = join(dirname(dirname(__file__)), 'example') client = ArrayClient(basepath) node = client.get_node('/pandashdf5/data') assert 'sample' in node.keys() node = node.get_node('sample') data = node.select() assert data.shape == (73, 2)
def test_hdf_node(): basepath = join(dirname(dirname(__file__)), 'example') client = ArrayClient(basepath) node = client.get_node('/pandashdf5/data') assert 'sample' in node.keys() node = node.get_node('sample') data = node.select() assert data.shape == (73,2)
def test_csv_node(): basepath = join(dirname(dirname(__file__)), 'example') client = ArrayClient(basepath) node = client.get_node('/customcsvs/sample') data1 = node.get() node = client.get_node('/customcsvs/sample2') data2 = node.select() node = client.get_node('/customcsvs/sample_pipe') data3 = node.select()
def test_custom_node(): basepath = join(dirname(dirname(__file__)), 'example') client = ArrayClient(basepath) node = client.get_node('/custom/sample2') data1 = node.select() node = client.get_node('/custom/sample') data2 = node.get() assert data2.iloc[2]['values'] == 2 assert data1.iloc[2]['values'] == 4
class ArrayManagementClient(object): """ Hangle ArrayManagement Connection """ def __init__(self): EstuarialDir = pjoin(expanduser('~'), '.estuarial') self.basedir = pjoin(os.path.dirname(__file__), '..', 'data', 'catalog', 'SQL_DATA') self.aclient = ArrayClient(basepath=self.basedir, localdatapath=EstuarialDir) self.aclient.set_logging(log)
def test_sql_yaml_cache(): basepath = join(dirname(dirname(__file__)), 'example') client = ArrayClient(basepath) arr = client['/sqlviews/example_sql.yaml'] date_1 = dt.datetime(2000, 1, 1) date_2 = dt.datetime(2003, 12, 30) aapl = arr.select(and_(arr.ticker == 'AAPL'), date_1=date_1, date_2=date_2) query = arr.select(and_(arr.ticker == 'AAPL'), date_1=date_1, date_2=date_2, IgnoreCache=True) query = arr.select(and_(arr.ticker == 'AAPL'), IgnoreCache=True) arr = client['/sqlviews/example_no_dates.yaml'] aapl = arr.select(and_(arr.ticker.in_(['A', 'AA']))) print aapl arr = client['/sqlviews/example_no_dates_not_entities.yaml'] aapl = arr.select(query_filter=None) print aapl query = arr.select(query_filter=None, IgnoreCache=True)
def __init__(self, data_directory): self.data_directory = data_directory try: from arraymanagement.client import ArrayClient self.client = ArrayClient(self.data_directory, configname="bokeh.server.hdf5_backend_config") except Exception as e: logger.exception(e) logger.info("error importing arraymanagement") logger.info("install arraymanagement from https://github.com/continuumio/ArrayManagement") logger.info("or procede without remote data capabilities")
def teardown_module(): basepath = join(dirname(dirname(__file__)), 'example') client = ArrayClient(basepath) client.clear_disk_cache()
def setup_module(): basepath = join(dirname(dirname(__file__)), 'example') client = ArrayClient(basepath) client.clear_disk_cache()