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)
