class TestFred(unittest.TestCase):
def setUp(self):
self.fred = Fred()
def testGetSeries(self):
s = self.fred.get_series('SP500', observation_start='9/2/2014', observation_end='9/5/2014')
self.assertEqual(s.ix['9/2/2014'], 2002.28)
self.assertEqual(len(s), 4)
info = self.fred.get_series_info('PAYEMS')
self.assertEqual(info['title'], 'All Employees: Total nonfarm')
# invalid series id
self.assertRaises(ValueError, self.fred.get_series, 'invalid')
self.assertRaises(ValueError, self.fred.get_series_info, 'invalid')
# invalid parameter
try:
self.fred.get_series('SP500', observation_start='invalid-datetime-str')
self.assertTrue(False, 'previous line should have thrown a ValueError')
except ValueError:
pass
def testSearch(self):
personal_income_series = self.fred.search_by_release(175, limit=3, order_by='popularity', sort_order='desc')
series_ids = ['PCPI06037', 'PCPI06075', 'PCPI24510']
for series_id in series_ids:
self.assertTrue(series_id in personal_income_series.index)
self.assertEqual(personal_income_series.ix[series_id, 'observation_start'], datetime(1969, 1, 1))
def tearDown(self):
return
class TestFred(unittest.TestCase):
def setUp(self):
self.fred = Fred()
def testGetSeries(self):
s = self.fred.get_series("SP500", observation_start="9/2/2014", observation_end="9/5/2014")
self.assertEqual(s.ix["9/2/2014"], 2002.28)
self.assertEqual(len(s), 4)
info = self.fred.get_series_info("PAYEMS")
self.assertEqual(info["title"], "All Employees: Total nonfarm")
# invalid series id
self.assertRaises(ValueError, self.fred.get_series, "invalid")
self.assertRaises(ValueError, self.fred.get_series_info, "invalid")
# invalid parameter
try:
self.fred.get_series("SP500", observation_start="invalid-datetime-str")
self.assertTrue(False, "previous line should have thrown a ValueError")
except ValueError:
pass
def testSearch(self):
personal_income_series = self.fred.search_by_release(175, limit=3, order_by="popularity", sort_order="desc")
series_ids = ["PCPI06037", "PCPI06075", "PCPI34039"]
for series_id in series_ids:
self.assertTrue(series_id in personal_income_series.index)
self.assertEqual(personal_income_series.ix[series_id, "observation_start"], datetime(1969, 1, 1))
def tearDown(self):
return
class TestFred(unittest.TestCase):
def setUp(self):
self.fred = Fred()
def testGetSeries(self):
s = self.fred.get_series('SP500',
observation_start='9/2/2014',
observation_end='9/5/2014')
self.assertEqual(s.ix['9/2/2014'], 2002.28)
self.assertEqual(len(s), 4)
info = self.fred.get_series_info('PAYEMS')
self.assertEqual(info['title'], 'All Employees: Total nonfarm')
# invalid series id
self.assertRaises(ValueError, self.fred.get_series, 'invalid')
self.assertRaises(ValueError, self.fred.get_series_info, 'invalid')
# invalid parameter
try:
self.fred.get_series('SP500',
observation_start='invalid-datetime-str')
self.assertTrue(False,
'previous line should have thrown a ValueError')
except ValueError:
pass
def testSearch(self):
personal_income_series = self.fred.search_by_release(
175, limit=3, order_by='popularity', sort_order='desc')
series_ids = ['PCPI06037', 'PCPI06075', 'PCPI34039']
for series_id in series_ids:
self.assertTrue(series_id in personal_income_series.index)
self.assertEqual(
personal_income_series.ix[series_id, 'observation_start'],
datetime(1969, 1, 1))
def tearDown(self):
return
# -*- coding: utf-8 -*-
"""
Created on Tue Jul 21 14:40:39 2015
@author: justin.malinchak
"""
from fredapi import Fred
fred = Fred()
#s = fred.get_series('DSPIC96', observation_start='1960-01-01', observation_end='2015-07-01')
#print s
df_homesales = fred.search_by_release(291,
limit=50,
order_by='popularity',
sort_order='desc')
print df_homesales['title']
df = {}
df['s1'] = fred.get_series('HOSMEDUSM052N')
df['s2'] = fred.get_series('HOSMEDUSNEM052N')
df['s3'] = fred.get_series('HOSMEDUSSOM052N')
import pandas as pd
df = pd.DataFrame(df)
df.plot()
print df['s1']
from fredapi import Fred
import pandas as pd
from chorogrid import Colorbin, Chorogrid
import os
import numpy as np
from scipy.stats.stats import pearsonr
####### Import the FRED Data #######
api_key = 'f213b90a3ec4042de2259bceceb6ccfa'
os.environ["FRED_API_KEY"] = api_key
apiFred = Fred()
dfFred = apiFred.search_by_release(112)
dfStateUnEmp = dfFred[dfFred['title'].str.contains("Unemployment Rate in ")&~dfFred['title'].str.contains("Census")&dfFred['seasonal_adjustment_short'].str.match('SA')&~dfFred['title'].str.contains('DISCONTINUED')]
srsStateUnIDs = dfStateUnEmp['id'][:]
dictStateUN = {}
for i in srsStateUnIDs:
dictStateUN[i[:2]] = apiFred.get_series(i,observation_start='2007-01-01').mean()
dfStateUN = pd.DataFrame(list(dictStateUN.items()),columns=['State', 'Unemployment'])
##Make the Map
colors = ['#fff5eb', '#fee6ce', '#fdd0a2', '#fdae6b', '#fd8d3c', '#f16913', '#d94801', '#8c2d04']
chBin = Colorbin(dfStateUN['Unemployment'], colors, proportional=True, decimals=None)
from fredapi import Fred
import pandas as pd
from chorogrid import Colorbin, Chorogrid
import os
import numpy as np
from scipy.stats.stats import pearsonr
####### Import the FRED Data #######
api_key = 'f213b90a3ec4042de2259bceceb6ccfa'
os.environ["FRED_API_KEY"] = api_key
apiFred = Fred()
dfFred = apiFred.search_by_release(112)
dfStateUnEmp = dfFred[dfFred['title'].str.contains("Unemployment Rate in ")
& ~dfFred['title'].str.contains("Census")
& dfFred['seasonal_adjustment_short'].str.match('SA')
& ~dfFred['title'].str.contains('DISCONTINUED')]
srsStateUnIDs = dfStateUnEmp['id'][:]
dictStateUN = {}
for i in srsStateUnIDs:
dictStateUN[i[:2]] = apiFred.get_series(
i, observation_start='2007-01-01').mean()
dfStateUN = pd.DataFrame(list(dictStateUN.items()),
columns=['State', 'Unemployment'])
##Make the Map
