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_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)