def test_update_datasource(token, dataset_meta, dataset_data):
client = Client(api_key=token, host="localhost:8030/api2", ssl=False)
datasources = client.list_datasources()
primary_key = datasources['results'][0]['pk']
datasource = client.get_datasource(primary_key)
meta = dataset_meta
meta['info']['from_source']['pandas_reader'] = 'changed'
data = dataset_data[:90].to_csv()
resp = datasource.update_meta_and_data(meta=meta, data=data)
datasource2 = client.get_datasource(primary_key)
meta_and_data = datasource2.get_meta_and_data()
new_meta = meta_and_data['meta']
new_data = meta_and_data['data']
new_data_df = pd.read_csv(StringIO(new_data))
assert new_meta['info']['from_source']['pandas_reader'] == 'changed'
assert resp.status_code == 200
def test_get_meta_and_data(token):
client = Client(api_key=token, host="localhost:8030/api2", ssl=False)
datasources = client.list_datasources()
primary_key = datasources['results'][0]['pk']
datasource = client.get_datasource(primary_key)
resp = datasource.get_meta_and_data()
print(resp.keys())
assert 'meta' in resp
assert 'data' in resp
def test_get_combined_tables(token):
client = Client(api_key=token, host="localhost:8030/api2", ssl=False)
datasources = client.list_datasources()
primary_key = datasources['results'][0]['pk']
datasource = client.get_datasource(primary_key)
tables = datasource.get_tables(['price', 'quality'], ['gender', 'agecat'],
['counts', 'c%', 'cbase', 'stddev'],
combine=True,
language='en-GB')
assert tables.shape == (24, 7)
def test_get_table_set(token):
client = Client(api_key=token, host="localhost:8030/api2", ssl=False)
datasources = client.list_datasources()
primary_key = datasources['results'][0]['pk']
datasource = client.get_datasource(primary_key)
stubs = [['distance', 'store', 'contact'], ['reason1', 'reason2', 'dept'],
['price', 'numitems', 'org', 'service', 'quality', 'overall']]
banners = [['gender', 'agecat'], ['regular', 'purchase']]
table_set = datasource.get_table_set(stubs, banners,
['base', 'counts', 'c%', 'stddev'])
assert len(table_set) == len(stubs) * len(banners)
def test_dataset_to_excel(token):
client = Client(api_key=token, host="localhost:8030/api2", ssl=False)
datasources = client.list_datasources()
primary_key = datasources['results'][0]['pk']
datasource = client.get_datasource(primary_key)
stubs = [['distance', 'store', 'contact'], ['reason1', 'reason2', 'dept'],
['price', 'numitems', 'org', 'service', 'quality', 'overall']]
banners = [['gender', 'agecat'], ['regular', 'purchase']]
table_set = datasource.get_table_set(stubs, banners,
['base', 'counts', 'c%', 'stddev'])
datasource.table_set_to_excel(table_set, 'myexcel.xlsx')
assert os.path.isfile('myexcel.xlsx')
os.remove("myexcel.xlsx")
def test_get_datasource(token):
client = Client(api_key=token, host="localhost:8030/api2", ssl=False)
datasources = client.list_datasources()
primary_key = datasources['results'][0]['pk']
resp = client.get_datasource(primary_key)
assert isinstance(resp, Datasource)
