MAINT: Refactors out selection of treasury duration its own function.

Instead of having the duration selection logic in choose_treasury,
break out duration specific logig into another function.
This commit is contained in:
Eddie Hebert
2013-04-03 15:36:01 -04:00
parent 0cc953e00f
commit cf60eeb46b
+7 -2
View File
@@ -201,7 +201,7 @@ def search_day_distance(end_date, dt):
return tdd
def choose_treasury(treasury_curves, start_date, end_date):
def select_treasury_duration(start_date, end_date):
td = end_date - start_date
if td.days <= 31:
treasury_duration = '1month'
@@ -224,6 +224,11 @@ def choose_treasury(treasury_curves, start_date, end_date):
else:
treasury_duration = '30year'
return treasury_duration
def choose_treasury(treasury_curves, start_date, end_date):
treasury_duration = select_treasury_duration(start_date, end_date)
end_day = end_date.replace(hour=0, minute=0, second=0)
search_day = None
@@ -262,7 +267,7 @@ treasury history range."
log.warn(message)
if search_day:
treasury_curves[search_day]
td = end_date - start_date
return rate * (td.days + 1) / 365
message = "No rate for end date = {dt} and term = {term}. Check \