def test_export_downloadable_csv(self, mock_bake): run_export(prep_only=True) export_downloadable_csv('County', 'percent_30_60') self.assertEqual(mock_bake.call_count, 1) export_downloadable_csv('County', 'percent_90') self.assertEqual(mock_bake.call_count, 2) export_downloadable_csv('MetroArea', 'percent_30_60') self.assertEqual(mock_bake.call_count, 3) export_downloadable_csv('MetroArea', 'percent_90') self.assertEqual(mock_bake.call_count, 4) export_downloadable_csv('State', 'percent_30_60') self.assertEqual(mock_bake.call_count, 5) export_downloadable_csv('State', 'percent_90') self.assertEqual(mock_bake.call_count, 6)
def test_run_export(self, mock_export): run_export() self.assertEqual(mock_export.call_count, 6)