mirror of
https://github.com/bspeice/metrik
synced 2024-11-04 22:48:11 -05:00
Add USD LIBOR rates
Also shift versions lower to what I'm actually testing on
This commit is contained in:
parent
6c91c1cb1f
commit
4dd3315d2c
42
metrik/tasks/ice.py
Normal file
42
metrik/tasks/ice.py
Normal file
@ -0,0 +1,42 @@
|
||||
from luigi.task import Task
|
||||
# noinspection PyUnresolvedReferences
|
||||
from six.moves.urllib.parse import quote_plus
|
||||
import pandas as pd
|
||||
import pytz
|
||||
from dateutil.parser import parse
|
||||
|
||||
|
||||
class USDLibor(Task):
|
||||
|
||||
@staticmethod
|
||||
def retrieve_data(date):
|
||||
url = ('https://www.theice.com/marketdata/reports/icebenchmarkadmin/'
|
||||
'ICELiborHistoricalRates.shtml?excelExport='
|
||||
'&criteria.reportDate={}&criteria.currencyCode=USD').format(
|
||||
quote_plus(date.strftime('%m/%d/%y'))
|
||||
)
|
||||
|
||||
def parse_london(dt_str):
|
||||
# Pandas does its best to try and help us out by modifying the
|
||||
# actual csv content to try and add timezone and date information.
|
||||
# Which is not in any sense what we want.
|
||||
# So we have convoluted steps to go and fix that.
|
||||
london_tz = pytz.timezone('Europe/London')
|
||||
# Note that parse() implicitly adds timezone information because
|
||||
# of how pandas gave us the value
|
||||
dt = parse(dt_str).replace(year=date.year,
|
||||
month=date.month,
|
||||
day=date.day)
|
||||
return dt.astimezone(london_tz)
|
||||
|
||||
# Skip 1 row at top for header (header=0),
|
||||
# and read 7 total rows. For whatever reason,
|
||||
# pandas totally ignores both skipfooter and skip_footer.
|
||||
# WTF pandas.
|
||||
df = pd.read_csv(
|
||||
url, names=['Tenor', 'Publication Time', 'USD ICE LIBOR'],
|
||||
header=0, parse_dates=['Publication Time'],
|
||||
nrows=7, date_parser=parse_london,
|
||||
)
|
||||
|
||||
return df
|
@ -1,3 +1,5 @@
|
||||
pytz>=2016.6.1
|
||||
luigi>=2.2.0
|
||||
pyquery>=1.2.13
|
||||
requests>=2.11.0
|
||||
pyquery>=1.2.9
|
||||
requests>=2.9.1
|
||||
pandas>=0.17.1
|
59
test/tasks/test_ice.py
Normal file
59
test/tasks/test_ice.py
Normal file
@ -0,0 +1,59 @@
|
||||
from unittest import TestCase
|
||||
from datetime import datetime
|
||||
import pytz
|
||||
|
||||
from metrik.tasks.ice import USDLibor
|
||||
|
||||
|
||||
# noinspection PyUnresolvedReferences
|
||||
class TestICE(TestCase):
|
||||
|
||||
def test_correct_libor_Aug8_2016(self):
|
||||
# Validate with:
|
||||
# https://www.theice.com/marketdata/reports/icebenchmarkadmin/ICELiborHistoricalRates.shtml?excelExport=&criteria.reportDate=8%2F8%2F16&criteria.currencyCode=USD
|
||||
aug8_libor = USDLibor.retrieve_data(datetime(2016, 8, 8))
|
||||
|
||||
assert (aug8_libor[aug8_libor['Tenor'] == 'Overnight']['USD ICE LIBOR'] == .4189).all()
|
||||
assert (aug8_libor[aug8_libor['Tenor'] == '1 Week']['USD ICE LIBOR'] == .4431).all()
|
||||
assert (aug8_libor[aug8_libor['Tenor'] == '1 Month']['USD ICE LIBOR'] == .5119).all()
|
||||
assert (aug8_libor[aug8_libor['Tenor'] == '2 Month']['USD ICE LIBOR'] == .6268).all()
|
||||
assert (aug8_libor[aug8_libor['Tenor'] == '3 Month']['USD ICE LIBOR'] == .8065).all()
|
||||
assert (aug8_libor[aug8_libor['Tenor'] == '6 Month']['USD ICE LIBOR'] == 1.1852).all()
|
||||
assert (aug8_libor[aug8_libor['Tenor'] == '1 Year']['USD ICE LIBOR'] == 1.5081).all()
|
||||
|
||||
london_tz = pytz.timezone('Europe/London')
|
||||
actual = london_tz.localize(datetime(2016, 8, 8, 11, 45, 6))
|
||||
assert (aug8_libor['Publication Time'] == actual).all()
|
||||
|
||||
def test_correct_libor_Aug9_2010(self):
|
||||
# Validate with:
|
||||
# https://www.theice.com/marketdata/reports/icebenchmarkadmin/ICELiborHistoricalRates.shtml?excelExport=&criteria.reportDate=8%2F9%2F10&criteria.currencyCode=USD
|
||||
aug9_libor = USDLibor.retrieve_data(datetime(2010, 8, 9))
|
||||
|
||||
assert (aug9_libor[aug9_libor['Tenor'] == 'Overnight']['USD ICE LIBOR'] == .23656).all()
|
||||
assert (aug9_libor[aug9_libor['Tenor'] == '1 Week']['USD ICE LIBOR'] == .27725).all()
|
||||
assert (aug9_libor[aug9_libor['Tenor'] == '1 Month']['USD ICE LIBOR'] == .29).all()
|
||||
assert (aug9_libor[aug9_libor['Tenor'] == '2 Month']['USD ICE LIBOR'] == .3375).all()
|
||||
assert (aug9_libor[aug9_libor['Tenor'] == '3 Month']['USD ICE LIBOR'] == .40438).all()
|
||||
assert (aug9_libor[aug9_libor['Tenor'] == '6 Month']['USD ICE LIBOR'] == .6275).all()
|
||||
assert (aug9_libor[aug9_libor['Tenor'] == '1 Year']['USD ICE LIBOR'] == .995).all()
|
||||
|
||||
london_tz = pytz.timezone('Europe/London')
|
||||
actual = london_tz.localize(datetime(2010, 8, 9, 15, 49, 12))
|
||||
assert (aug9_libor['Publication Time'] == actual).all()
|
||||
|
||||
def test_correct_date_reasoning(self):
|
||||
# Make sure I document how to handle datetime issues in the future
|
||||
london_tz = pytz.timezone('Europe/London')
|
||||
ny_tz = pytz.timezone('America/New_York')
|
||||
|
||||
# DON'T YOU DARE SET TZINFO, SHENANIGANS HAPPEN
|
||||
assert (datetime(2016, 8, 8, 15, 0, 0, tzinfo=london_tz) <
|
||||
datetime(2016, 8, 8, 15, 0, 0, tzinfo=ny_tz))
|
||||
|
||||
assert (datetime(2016, 8, 8, 15, 0, 0, tzinfo=london_tz) >
|
||||
datetime(2016, 8, 8, 10, 0, 0, tzinfo=ny_tz))
|
||||
|
||||
# ALWAYS USE timezone.localize()
|
||||
assert (london_tz.localize(datetime(2016, 8, 8, 15)) ==
|
||||
ny_tz.localize(datetime(2016, 8, 8, 10)))
|
Loading…
Reference in New Issue
Block a user