def test_members():
set_default_client(None)
assert default_client() is None
c = Client(NEUPRINT_SERVER, DATASET)
assert c.server == f'https://{NEUPRINT_SERVER}'
assert c.dataset == DATASET
assert default_client() is c
df = c.fetch_custom("MATCH (m:Meta) RETURN m.primaryRois as rois")
assert isinstance(df, pd.DataFrame)
assert df.columns == ['rois']
assert len(df) == 1
assert isinstance(df['rois'].iloc[0], list)
assert isinstance(c.fetch_available(), list)
assert isinstance(c.fetch_help(), str)
assert c.fetch_server_info() is True
assert isinstance(c.fetch_version(), str)
assert isinstance(c.fetch_database(), dict)
assert isinstance(c.fetch_datasets(), dict)
assert isinstance(c.fetch_db_version(), str)
assert isinstance(c.fetch_profile(), dict)
assert isinstance(c.fetch_token(), str)
assert isinstance(c.fetch_daily_type(), tuple)
assert isinstance(c.fetch_roi_completeness(), pd.DataFrame)
assert isinstance(c.fetch_roi_connectivity(), pd.DataFrame)
assert isinstance(c.fetch_roi_mesh('AB(R)'), bytes)
assert isinstance(c.fetch_skeleton(EXAMPLE_BODY), pd.DataFrame)
def fetch_neuprint_bodies(dataset):
""" Create a dictionary (keyed by body ID) and set of bodies from NeuPrint
Keyword arguments:
dataset: data set
Returns:
result: dictionary of rows from NeuPrint
neuprint_bodyset: set of body IDs
"""
query = """
MATCH (n: Neuron)
RETURN n.bodyId as bodyId,n.status as status,n.statusLabel as label,n.type as type,n.instance as instance,n.size as size
ORDER BY n.type, n.instance
"""
print(default_client())
results = fetch_custom(query, format='json')
LOGGER.info("%d Body IDs found in NeuPrint %s", len(results['data']),
dataset)
neuprint_bodyset = set()
for row in results['data']:
neuprint_bodyset.add(str(row[0]))
return results['data'], neuprint_bodyset
def client():
c = Client(NEUPRINT_SERVER, DATASET)
set_default_client(c)
assert default_client() is c
return c
