Ejemplo n.º 1
0
    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')
Ejemplo n.º 2
0
    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')
Ejemplo n.º 3
0
    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)
Ejemplo n.º 4
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))
Ejemplo n.º 5
0
    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)
Ejemplo n.º 6
0
Archivo: test.py Proyecto: jmvg/pyjstat
    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)))
Ejemplo n.º 7
0
 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)))
Ejemplo n.º 8
0
Archivo: test.py Proyecto: jmvg/pyjstat
    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))
Ejemplo n.º 9
0
Archivo: test.py Proyecto: jmvg/pyjstat
    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))
Ejemplo n.º 10
0
 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))
Ejemplo n.º 11
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)))
Ejemplo n.º 12
0
Archivo: test.py Proyecto: jmvg/pyjstat
    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))
Ejemplo n.º 13
0
Archivo: test.py Proyecto: jmvg/pyjstat
    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))
Ejemplo n.º 14
0
 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))
Ejemplo n.º 15
0
 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))
Ejemplo n.º 16
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))
Ejemplo n.º 17
0
    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))
Ejemplo n.º 18
0
    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))
Ejemplo n.º 19
0
Archivo: test.py Proyecto: jmvg/pyjstat
    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))
Ejemplo n.º 20
0
 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))
Ejemplo n.º 21
0
    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))
Ejemplo n.º 22
0
Archivo: test.py Proyecto: jmvg/pyjstat
    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")
Ejemplo n.º 23
0
Archivo: test.py Proyecto: jmvg/pyjstat
    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)