Beispiel #1
0
 def setUp(self):
     self.store = OECDStore()
Beispiel #2
0
class TestOECDTestSite(tm.TestCase):

    def setUp(self):
        self.store = OECDStore()

    def test_isvalid(self):
        self.assertTrue(self.store.is_valid())

    def test_get_un_den(self):
        data = self.store.get('UN_DEN')
        self.assertTrue(isinstance(data, OECDResource))

        df = data.read()

        au = [50.17292785, 49.47181009, 49.52106174, 49.16341327, 48.19296375,
              47.8863461, 45.83517292, 45.02021403, 44.78983834, 44.37794217,
              44.15358142, 45.38865546, 46.33092037, 47.2343406, 48.80023876,
              50.0639872, 50.23390644, 49.8214994, 49.67636585, 49.55227375,
              48.48657368, 47.41179739, 47.52526561, 47.93048854, 47.26327162,
              45.4617105, 43.90202112, 42.32759607, 40.35838899, 39.35157364,
              39.55023059, 39.93196617, 39.21915611, 37.24341973, 34.42587311,
              32.51238056, 31.16811321, 29.78802692, 28.14678656, 25.41919614,
              25.71802695, 24.53149132, 23.21981032, 22.99175908, 22.29392791,
              22.29118237, 20.22236361, 18.51151852, 18.5674024, 19.31219498,
              18.44405734, 18.51407731, 18.19718895, 17.04071205]
        jp = [32.89493671, 33.80225989, 34.5969919, 35.01871257, 35.46869345,
              35.28164117, 34.749499, 34.40573103, 34.50762389, 35.16411379,
              35.10284332, 34.57209848, 34.31168831, 33.46611342, 34.26450371,
              34.53099287, 33.69881466, 32.99814274, 32.59541985, 31.75696594,
              31.14832536, 30.8917513, 30.56612982, 29.75285171, 29.22391559,
              28.79202411, 28.18680064, 27.71454381, 26.94358748, 26.13165206,
              26.13236815, 25.24310276, 24.98554405, 24.83467897, 24.7263178,
              24.38077142, 23.3953401, 22.78797997, 22.52794337, 22.18157944,
              21.54406273, 20.88284597, 20.26073907, 19.73945642, 19.06442577,
              18.79844243, 18.3497807, 18.25095057, 18.2204924, 18.45787546,
              18.38733297, 18.99504195, 17.97238372, 17.78318026]
        us = [30.89748411, 29.51891217, 29.34276869, 28.51337535, 28.30646144,
              28.16661991, 28.19557735, 27.76578899, 27.9004622, 27.30836054,
              27.43402867, 26.94941363, 26.25996487, 25.83134349, 25.74427582,
              25.28771204, 24.38412814, 23.59186681, 23.94328194, 22.3651229,
              22.06009466, 21.01328205, 20.47463895, 19.45290876, 18.22953818,
              17.44855678, 17.00126975, 16.5162476, 16.24744487, 15.86401127,
              15.45147174, 15.46986912, 15.1499578, 15.13654544, 14.91544059,
              14.31762091, 14.02052225, 13.55213736, 13.39571457, 13.36670812,
              12.90865079, 12.86997731, 12.76906383, 12.39142968, 12.02130767,
              11.96023574, 11.48458378, 11.56435375, 11.91022276, 11.79401904,
              11.38345975, 11.32948829, 11.07884758, 10.80789137]

        index = pd.DatetimeIndex(map(str, range(1960, 2014)))
        for label, values in [('Australia', au), ('Japan', jp), ('United States', us)]:
            expected = pd.Series(values, index=index)
            tm.assert_series_equal(df[label], expected)

        raw_data = data.read(raw=True)
        self.assertTrue(len(raw_data) > 0)

    def test_get_tourism(self):
        data = self.store.get('TOURISM_INBOUND')
        df = data.read()

        jp = np.array([6138, 6728, 7334, 8347, 8351, 6790, 8611, 6219, 8368, ], dtype=float)
        us = np.array([np.nan, np.nan, 183178, 175142, 175632, 160359, 162269, 164672, 171630], dtype=float)

        index = pd.DatetimeIndex(['2004', '2005', '2006', '2007', '2008',
                                  '2009', '2010', '2011', '2012'])
        for label, values in [('Japan', jp), ('United States', us)]:
            expected = pd.Series(values, index=index)
            tm.assert_series_equal(df[label]['Total international arrivals'], expected)