def test_numpy_dtypes(self): self.assertEqual( cpi.get(pd.np.int64(1950)), cpi.get(1950) ) self.assertEqual( cpi.inflate(100, pd.np.int32(1950)), cpi.inflate(100, 1950), ) self.assertEqual( cpi.inflate(100, pd.np.int64(1950), to=pd.np.int64(1960)), cpi.inflate(100, 1950, to=1960), ) self.assertEqual( cpi.inflate(100, pd.np.int64(1950), to=pd.np.int32(1960)), cpi.inflate(100, 1950, to=1960), ) self.assertEqual( cpi.inflate(100, pd.np.int64(1950), to=1960), cpi.inflate(100, 1950, to=1960), ) self.assertEqual( cpi.inflate(100, pd.to_datetime("1950-07-01"), to=pd.to_datetime("1960-07-01")), cpi.inflate(100, date(1950, 7, 1), to=date(1960, 7, 1)) )
def test_get(self): self.assertEqual(cpi.get(CPITest.TEST_YEAR_EARLIER), 24.1) self.assertEqual(cpi.get(CPITest.TEST_YEAR_LATER), 172.2) with self.assertRaises(CPIDoesNotExist): cpi.get(1900) with self.assertRaises(CPIDoesNotExist): cpi.get(date(1900, 1, 1)) with self.assertRaises(CPIDoesNotExist): cpi.get(1950, series="FOOBAR")
def test_get_errors(self): with self.assertRaises(CPIObjectDoesNotExist): cpi.get(1900) with self.assertRaises(CPIObjectDoesNotExist): cpi.get(date(1900, 1, 1)) with self.assertRaises(CPIObjectDoesNotExist): cpi.get(1950, series_id="FOOBAR")
def test_get(self): self.assertEqual(cpi.get(1950), 24.1) self.assertEqual(cpi.get(2000), 172.2) with self.assertRaises(CPIDoesNotExist): cpi.get(1900) with self.assertRaises(CPIDoesNotExist): cpi.get(date(1900, 1, 1))
def test_get_by_series_id(self): self.assertEqual(cpi.get(date(1950, 1, 1), series_id="CUSR0000SA0"), 23.51)
def test_get_by_kwargs(self): # "CUUR0000SA0" self.assertEqual(cpi.get(2000), 172.2) # "CUSR0000SA0" self.assertEqual(cpi.get(date(2000, 1, 1), seasonally_adjusted=True), 169.30) # ... which doesn't have annual values with self.assertRaises(CPIObjectDoesNotExist): cpi.get(2000, seasonally_adjusted=True) # "CUSR0000SA0E" # ... which we don't have loaded yet as data with self.assertRaises(CPIObjectDoesNotExist): cpi.get(2000, seasonally_adjusted=True, items="Energy") # "CUURS49ASA0" self.assertEqual( cpi.get(2000, area="Los Angeles-Long Beach-Anaheim, CA"), 171.6) self.assertEqual( cpi.get(date(2000, 1, 1), area="Los Angeles-Long Beach-Anaheim, CA"), 167.9) # "CUURS49ASA0E" self.assertEqual( cpi.get(2000, items="Energy", area="Los Angeles-Long Beach-Anaheim, CA"), 132.0) # "CUURA421SAT" self.assertEqual( cpi.get(2000, items="Transportation", area="Los Angeles-Long Beach-Anaheim, CA"), 154.2) # "CUURA421SA0E" self.assertEqual( cpi.get(2000, items="All items - old base", area="Los Angeles-Long Beach-Anaheim, CA"), 506.8)
def test_get(self): self.assertEqual(cpi.get(1950), 24.1) self.assertEqual(cpi.get(date(1950, 1, 1)), 23.5) self.assertEqual(cpi.get(2000), 172.2)
def test_get_value_error(self): with self.assertRaises(ValueError): cpi.get(1900.1) cpi.get(datetime.now()) cpi.get(3000)