def test_run_aggregates(self, mock_load_national, mock_load_states, mock_load_msas, mock_load_fips, mock_load_non_msa): from data_research.scripts.load_mortgage_aggregates import FIPS FIPS.dates = ['2016-09-01'] run_aggregates() self.assertEqual(mock_load_fips.call_count, 1) self.assertEqual(mock_load_national.call_count, 1) self.assertEqual(mock_load_states.call_count, 1) self.assertEqual(mock_load_msas.call_count, 1)
def test_run_aggregates(self, mock_validate_counties, mock_update_dates): dates = MortgageMetaData.objects.get(name='sampling_dates') dates.json_value = ['2016-01-01'] dates.save() run_aggregates() self.assertEqual(mock_validate_counties.call_count, 1) self.assertEqual(mock_update_dates.call_count, 1) self.assertEqual(NationalMortgageData.objects.count(), 1) self.assertEqual(StateMortgageData.objects.count(), 1) self.assertEqual(MSAMortgageData.objects.count(), 1) self.assertEqual(NonMSAMortgageData.objects.count(), 1)
def test_run_aggregates(self, mock_validate_counties, mock_update_dates): dates = MortgageMetaData.objects.get( name='sampling_dates') dates.json_value = ['2016-01-01'] dates.save() run_aggregates() self.assertEqual(mock_validate_counties.call_count, 1) self.assertEqual(mock_update_dates.call_count, 1) self.assertEqual(NationalMortgageData.objects.count(), 1) self.assertEqual(StateMortgageData.objects.count(), 1) self.assertEqual(MSAMortgageData.objects.count(), 1) self.assertEqual(NonMSAMortgageData.objects.count(), 1)