def _mock_lcs_get_land_cover_region_wrong_dataset(id): dds = DrivingDataset() dds.id = 7 land_cover_type = LandCoverRegionCategory() land_cover_type.id = 1 land_cover_type.driving_dataset_id = 7 land_cover_type.driving_dataset = dds land_cover_region = LandCoverRegion() land_cover_region.id = id land_cover_region.name = "Sand" land_cover_region.category = land_cover_type return land_cover_region
def setUp(self): driving_data = DrivingDataset() driving_data.name = "Driving1" driving_data.id = 1 self.return_dd_id = 10 self.model_run = ModelRun() self.model_run.name = "Model Run1" self.model_run.driving_dataset = driving_data self.model_run.driving_dataset_id = 1 self.land_cover_service = LandCoverService() self.land_cover_service.get_land_cover_region_by_id = self._mock_lcs_get_land_cover_region self.land_cover_service.get_land_cover_values = self._mock_get_land_cover_values self.land_cover_service.get_land_cover_categories = self._mock_get_land_cover_categories self.land_cover_service.save_land_cover_actions_for_model = MagicMock() self.land_cover_service.get_land_cover_actions_for_model = self._mock_lcs_get_actions_for_model_run self.land_cover_service.save_fractional_land_cover_for_model = MagicMock() self.land_cover_service.get_default_fractional_cover = self._mock_lcs_get_default_fractional_cover self.land_cover_service.save_default_soil_properties = MagicMock() self.land_cover_helper = LandCoverControllerHelper(land_cover_service=self.land_cover_service)