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