def test_get_dimensions_by_label(self):
""" Test pyjstat get_dimensions() using label as parameter """
dimensions, dim_names = pyjstat.get_dimensions(
self.oecd_datasets['oecd'], 'label')
self.assertTrue(dim_names[2] == '2003-2014')
self.assertTrue(dimensions[0].iloc[0]['label'] == 'unemployment rate')
def test_get_dimensions_by_label(self):
""" Test pyjstat get_dimensions() using label as parameter """
dimensions, dim_names = pyjstat.get_dimensions(
self.oecd_datasets['oecd'], 'label')
self.assertTrue(dim_names[2] == '2003-2014')
self.assertTrue(dimensions[0].iloc[0]['label'] == 'Unemployment rate')
def test_get_dimensions_by_index(self):
""" Test pyjstat get_dimensions() using id as parameter """
dimensions, dim_names = pyjstat.get_dimensions(
self.oecd_datasets['oecd'], 'index')
self.assertTrue(dim_names[2] == 'year')
self.assertTrue(dimensions[0].iloc[0]['index'] == 0)
def test_generate_df_with_id(self):
"""Test pyjstat generate_df() using id as parameter."""
data_df = pyjstat.generate_df(self.oecd_datasets['oecd'], 'id')
line_thirty = ['UNR', 'BE', '2009', 7.891892855]
dimensions = pyjstat.get_dimensions(self.oecd_datasets['oecd'], 'id')
self.assertTrue(set(data_df.columns.values[:-1]) == set(dimensions[1]))
self.assertTrue(set(data_df.iloc[30].values) == set(line_thirty))
def test_get_dimensions_by_index(self):
""" Test pyjstat get_dimensions() using id as parameter """
dimensions, dim_names = pyjstat.get_dimensions(
self.oecd_datasets['oecd'], 'index')
self.assertTrue(dim_names[2] == 'year')
self.assertTrue(dimensions[0].iloc[0]['index'] == 0)
def test_get_df_row(self):
""" Test pyjstat get_df_row() """
dimensions = pyjstat.get_dimensions(self.oecd_datasets["oecd"], "label")
first_row = ["unemployment rate", "Australia", "2003"]
categories = pyjstat.get_df_row(dimensions[0])
self.assertTrue(set(first_row) == set(next(categories)))
def test_get_df_row(self):
"""Test pyjstat get_df_row().expandvars."""
dimensions = pyjstat.get_dimensions(self.oecd_datasets['oecd'],
'label')
first_row = ['unemployment rate', 'Australia', '2003']
categories = pyjstat.get_df_row(dimensions[0])
self.assertTrue(set(first_row) == set(next(categories)))
def test_generate_df_with_id(self):
""" Test pyjstat generate_df() using id as parameter"""
data_df = pyjstat.generate_df(self.oecd_datasets["oecd"], "id")
line_thirty = ["UNR", "BE", "2009", 7.891892855]
dimensions = pyjstat.get_dimensions(self.oecd_datasets["oecd"], "id")
self.assertTrue(set(data_df.columns.values[:-1]) == set(dimensions[1]))
self.assertTrue(set(data_df.iloc[30].values) == set(line_thirty))
def test_generate_df_with_label(self):
""" Test pyjstat generate_df() using label as parameter"""
data_df = pyjstat.generate_df(self.oecd_datasets["oecd"], "label")
line_thirty = ["unemployment rate", "Belgium", "2009", 7.891892855]
dimensions = pyjstat.get_dimensions(self.oecd_datasets["oecd"], "label")
self.assertTrue(set(data_df.columns.values[:-1]) == set(dimensions[1]))
self.assertTrue(set(data_df.iloc[30].values) == set(line_thirty))
def test_generate_df_with_label(self):
"""Test pyjstat generate_df() using label as parameter."""
data_df = pyjstat.generate_df(self.oecd_datasets['oecd'], 'label')
line_thirty = ['unemployment rate', 'Belgium', '2009', 7.891892855]
dimensions = pyjstat.get_dimensions(self.oecd_datasets['oecd'],
'label')
self.assertTrue(set(data_df.columns.values[:-1]) == set(dimensions[1]))
self.assertTrue(set(data_df.iloc[30].values) == set(line_thirty))
def test_get_df_row(self):
""" Test pyjstat get_df_row() """
dimensions = pyjstat.get_dimensions(self.oecd_datasets['oecd'],
'label')
first_row = ['Unemployment rate', 'Australia', '2003']
categories = pyjstat.get_df_row(dimensions[0])
self.assertTrue(set(first_row) == set(next(categories)))
def test_from_json_stat_with_id(self):
""" Test pyjstat from_json_stat() using id as parameter"""
results = pyjstat.from_json_stat(self.oecd_datasets, naming="id")
line_thirty = [u"UNR", u"BE", u"2009", 7.891892855]
dimensions = pyjstat.get_dimensions(self.oecd_datasets["oecd"], "id")
self.assertTrue(len(results) == 2)
self.assertTrue(set(results[0].columns.values[:-1]) == set(dimensions[1]))
self.assertTrue(set(results[0].iloc[30].values) == set(line_thirty))
def test_from_json_stat_with_label(self):
""" Test pyjstat from_json_stat() using label as parameter """
results = pyjstat.from_json_stat(self.oecd_datasets)
line_thirty = ["unemployment rate", "Belgium", "2009", 7.891892855]
dimensions = pyjstat.get_dimensions(self.oecd_datasets["oecd"], "label")
self.assertTrue(len(results) == 2)
self.assertTrue(set(results[0].columns.values[:-1]) == set(dimensions[1]))
self.assertTrue(set(results[0].iloc[30].values) == set(line_thirty))
def test_from_json_stat_with_id(self):
"""Test pyjstat from_json_stat() using id as parameter."""
results = pyjstat.from_json_stat(self.oecd_datasets, naming='id')
line_thirty = [u'UNR', u'BE', u'2009', 7.891892855]
dimensions = pyjstat.get_dimensions(self.oecd_datasets['oecd'], 'id')
self.assertTrue(len(results) == 2)
self.assertTrue(
set(results[0].columns.values[:-1]) == set(dimensions[1]))
self.assertTrue(set(results[0].iloc[30].values) == set(line_thirty))
def test_from_json_stat_with_label(self):
"""Test pyjstat from_json_stat() using label as parameter."""
results = pyjstat.from_json_stat(self.oecd_datasets)
line_thirty = ['unemployment rate', 'Belgium', '2009', 7.891892855]
dimensions = pyjstat.get_dimensions(self.oecd_datasets['oecd'],
'label')
self.assertTrue(len(results) == 2)
self.assertTrue(
set(results[0].columns.values[:-1]) == set(dimensions[1]))
self.assertTrue(set(results[0].iloc[30].values) == set(line_thirty))
def test_generate_df_with_id(self):
""" Test pyjstat generate_df() using id as parameter"""
data_df = pyjstat.generate_df(self.oecd_datasets['oecd'], 'id')
line_thirty = ['UNR', 'BE', 2009, 7.891892855]
dimensions = pyjstat.get_dimensions(self.oecd_datasets['oecd'], 'id')
self.assertTrue(set(data_df.columns.values[:-1]) ==
set(dimensions[1]))
self.assertTrue(set(data_df.iloc[30].values) ==
set(line_thirty))
def test_generate_df_with_label(self):
""" Test pyjstat generate_df() using label as parameter"""
data_df = pyjstat.generate_df(self.oecd_datasets['oecd'], 'label')
line_thirty = ['Unemployment rate', 'Belgium', 2009, 7.891892855]
dimensions = pyjstat.get_dimensions(self.oecd_datasets['oecd'],
'label')
self.assertTrue(set(data_df.columns.values[:-1]) ==
set(dimensions[1]))
self.assertTrue(set(data_df.iloc[30].values) ==
set(line_thirty))
def test_from_json_stat_with_id(self):
""" Test pyjstat from_json_stat() using id as parameter"""
results = pyjstat.from_json_stat(self.oecd_datasets, naming='id')
line_thirty = ['UNR', 'BE', 2009, 7.891892855]
dimensions = pyjstat.get_dimensions(self.oecd_datasets['oecd'], 'id')
self.assertTrue(len(results) == 2)
self.assertTrue(set(results[0].columns.values[:-1]) ==
set(dimensions[1]))
self.assertTrue(set(results[0].iloc[30].values) ==
set(line_thirty))
def test_from_to_json_stat_no_loads(self):
""" Test pyjstat nested from-to json_stat using list of dicts as input
"""
results = pyjstat.from_json_stat(self.oecd_datasets)
json_data = json.loads(pyjstat.to_json_stat(results), object_pairs_hook=OrderedDict)
data_df = pyjstat.from_json_stat(json_data)
line_thirty = ["unemployment rate", "Belgium", "2009", 7.891892855]
dimensions = pyjstat.get_dimensions(self.oecd_datasets["oecd"], "label")
self.assertTrue(len(data_df) == 2)
self.assertTrue(set(data_df[0].columns.values[:-1]) == set(dimensions[1]))
self.assertTrue(set(data_df[0].iloc[30].values) == set(line_thirty))
def test_from_to_json_stat_no_loads(self):
"""Test pyjstat nested from-to json_stat w list of dicts as input."""
results = pyjstat.from_json_stat(self.oecd_datasets)
json_data = json.loads(pyjstat.to_json_stat(results),
object_pairs_hook=OrderedDict)
data_df = pyjstat.from_json_stat(json_data)
line_thirty = ['unemployment rate', 'Belgium', '2009', 7.891892855]
dimensions = pyjstat.get_dimensions(self.oecd_datasets['oecd'],
'label')
self.assertTrue(len(data_df) == 2)
self.assertTrue(
set(data_df[0].columns.values[:-1]) == set(dimensions[1]))
self.assertTrue(set(data_df[0].iloc[30].values) == set(line_thirty))
def test_from_to_json_stat_as_dict(self):
""" Test pyjstat nested from-to json_stat using dict of dicts as input
"""
results = pyjstat.from_json_stat(self.oecd_datasets)
json_data = json.loads(pyjstat.to_json_stat(results, output='dict'),
object_pairs_hook=OrderedDict)
data_df = pyjstat.from_json_stat(
json.loads(json.dumps(json_data), object_pairs_hook=OrderedDict))
line_thirty = ['unemployment rate', 'Belgium', '2009', 7.891892855]
dimensions = pyjstat.get_dimensions(self.oecd_datasets['oecd'],
'label')
self.assertTrue(len(data_df) == 2)
self.assertTrue(set(data_df[0].columns.values[:-1]) ==
set(dimensions[1]))
self.assertTrue(set(data_df[0].iloc[30].values) ==
set(line_thirty))
def test_get_dimensions_by_label(self):
""" Test pyjstat get_dimensions() using label as parameter """
dimensions, dim_names = pyjstat.get_dimensions(self.oecd_datasets["oecd"], "label")
self.assertTrue(dim_names[2] == "2003-2014")
self.assertTrue(dimensions[0].iloc[0]["label"] == "unemployment rate")
def test_get_dimensions_by_index(self):
""" Test pyjstat get_dimensions() using id as parameter """
dimensions, dim_names = pyjstat.get_dimensions(self.oecd_datasets["oecd"], "index")
self.assertTrue(dim_names[2] == "year")
self.assertTrue(dimensions[0].iloc[0]["index"] == 0)