fix handling of latest year in tests

cfpb · Sep 27, 2016 · 318e1be · 318e1be
1 parent b57cdd3
commit 318e1be
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/paying_for_college/tests/test_scripts.py b/paying_for_college/tests/test_scripts.py
@@ -98,8 +98,8 @@ class TestScripts(django.test.TestCase):
                    'degrees_predominant': '',
                    'degrees_highest': '',
                    'school.ownership': 2,
-                   '2013.completion.completion_rate_4yr_150nt_pooled': 0.45,
-                   '2013.completion.completion_rate_less_than_4yr_150nt_pooled': None,
+                   '{}.completion.completion_rate_4yr_150nt_pooled'.format(YEAR): 0.45,
+                   '{}.completion.completion_rate_less_than_4yr_150nt_pooled'.format(YEAR): None,
                    'school.main_campus': True,
                    'school.online_only': False,
                    'school.operating': True,
@@ -124,8 +124,8 @@ class TestScripts(django.test.TestCase):
                    'degrees_predominant': '',
                    'degrees_highest': '',
                    'school.ownership': 2,
-                   '2013.completion.completion_rate_4yr_150nt_pooled': 0,
-                   '2013.completion.completion_rate_less_than_4yr_150nt_pooled': 0.25,
+                   '{}.completion.completion_rate_4yr_150nt_pooled'.format(YEAR): 0,
+                   '{}.completion.completion_rate_less_than_4yr_150nt_pooled'.format(YEAR): 0.25,
                    'school.main_campus': True,
                    'school.online_only': False,
                    'school.operating': False,
@@ -431,8 +431,8 @@ def test_calculate_percent(self):
     def test_get_repayment_data(self, mock_requests):
         mock_response = mock.Mock()
         expected_dict = {'results':
-                         [{'2013.repayment.5_yr_repayment.completers': 100,
-                          '2013.repayment.5_yr_repayment.noncompleters': 900}]}
+                         [{'{}.repayment.5_yr_repayment.completers'.format(YEAR): 100,
+                          '{}.repayment.5_yr_repayment.noncompleters'.format(YEAR): 900}]}
         mock_response.json.return_value = expected_dict
         mock_requests.return_value = mock_response
         data = api_utils.get_repayment_data(123456, YEAR)