def test_is_possible_returns_false_if_the_resource_is_not_harvestable_from_the_units_current_terrain(self): # noqa dummy_resource = MagicMock() dummy_resource.harvestable_from = set() dummy_unit = MagicMock() dummy_unit.tile.terrain = 'dummy_terrain' harvest = Harvest(dummy_resource, 1) self.assertFalse(harvest.is_possible(dummy_unit, 1.0))
def test_is_possible_returns_true_if_the_units_can_harvest_the_target_resource(self): # noqa dummy_resource = MagicMock() dummy_resource.harvestable_from = set(['dummy_terrain']) dummy_unit = MagicMock() dummy_unit.tile.terrain = 'dummy_terrain' dummy_unit.container.remaining_capacity.return_value = 1.0 harvest = Harvest(dummy_resource, 1) self.assertTrue(harvest.is_possible(dummy_unit, 1.0))