def test_lookup_median(self, calc, LendingStats):
lender_str = self.respondent.institution_id
# No lender
self.assertEqual(None, lookup_median(None, None))
# All of the US
lookup_median(self.respondent, None)
self.assertEqual(calc.call_args[0], (lender_str, None))
# Entry in the db
mock_obj = Mock()
mock_obj.lar_median = 9898
LendingStats.objects.filter.return_value.first.return_value = mock_obj
self.assertEqual(9898, lookup_median(self.respondent, self.metro))
# No entry in db
LendingStats.objects.filter.return_value.first.return_value = None
lookup_median(self.respondent, self.metro)
self.assertEqual(calc.call_args[0], (lender_str, self.metro))
def test_lookup_median(self, calc, LendingStats):
lender_str = self.respondent.institution_id
# No lender
self.assertEqual(None, lookup_median(None, None))
# All of the US
lookup_median(self.respondent, None)
self.assertEqual(calc.call_args[0], (lender_str, None))
# Entry in the db
mock_obj = Mock()
mock_obj.lar_median = 9898
LendingStats.objects.filter.return_value.first.return_value = mock_obj
self.assertEqual(9898, lookup_median(self.respondent, self.metro))
# No entry in db
LendingStats.objects.filter.return_value.first.return_value = None
lookup_median(self.respondent, self.metro)
self.assertEqual(calc.call_args[0], (lender_str, self.metro))