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