def mca(mock_imports): _, MCA = mock_imports ju.read_url_get = \ Mock(name='read_url_get', return_value={'msg': _msg}) ju.write = \ Mock(name='write') ju.read = \ Mock(name='read', return_value=_pd_msg) pj.read_json = \ Mock(name='read_json', return_value=_pd_msg) pd.DataFrame.to_csv = \ Mock(name='to_csv') pd.DataFrame.read_csv = \ Mock(name='read_csv', return_value=_csv_msg) Manifest.safe_mkdir = Mock(name='safe_mkdir') mca = MCA() mca.retrieve_file_over_http = Mock(name='retrieve_file_over_http') return mca
def connectivity(): nrrd.read = MagicMock(name='nrrd_read_file', return_value=('mock_annotation_data', 'mock_annotation_image')) conn_api = MouseConnectivityApi() download_link = '/path/to/link' conn_api.do_query = MagicMock(return_value=download_link) conn_api.json_msg_query = MagicMock(name='json_msg_query') conn_api.retrieve_file_over_http = \ MagicMock(name='retrieve_file_over_http') return conn_api