def test_digest_does_nothing_if_the_cargo_container_cannot_unload_requirements_or_load_products(self): # noqa tray = SlottedCargoContainer() tray.add_resource_slot(Fish, 10) tray.add_resource_slot(Wood, 10) tray.add_resource_slot(Cabbage, 10) tray.load_cargo(Fish, 3) tray.load_cargo(Wood, 5) tray.load_cargo(Cabbage, 4) plant1 = ResourcePlant() plant1.add_resource_requirement(Fish, 4) plant1.add_resource_requirement(Wood, 2) plant1.add_resource_product(Cabbage, 6) plant1.digest(tray) self.assertEqual(tray.cargo_slots[Fish.name]['load'], 3) self.assertEqual(tray.cargo_slots[Wood.name]['load'], 5) self.assertEqual(tray.cargo_slots[Cabbage.name]['load'], 4) plant2 = ResourcePlant() plant2.add_resource_requirement(Fish, 3) plant2.add_resource_requirement(Wood, 2) plant2.add_resource_product(Cabbage, 7) plant2.digest(tray) self.assertEqual(tray.cargo_slots[Fish.name]['load'], 3) self.assertEqual(tray.cargo_slots[Wood.name]['load'], 5) self.assertEqual(tray.cargo_slots[Cabbage.name]['load'], 4)
def __init__(self): super().__init__() self.container.add_resource_slot(Cabbage, 5) self.container.add_resource_slot(Wood, 5) cabbage_plant = ResourcePlant() cabbage_plant.add_resource_requirement(Wood, 1) cabbage_plant.add_resource_product(Cabbage, 1) self.add_resource_plant(cabbage_plant)
def __init__(self): super().__init__() self.container.add_resource_slot(Wood, 5) self.container.add_resource_slot(Lumber, 5) self.add_producable_resource(Lumber) lumber_plant = ResourcePlant() lumber_plant.add_resource_requirement(Wood, 1) lumber_plant.add_resource_product(Lumber, 1) self.add_resource_plant(lumber_plant)
def test_is_possible_returns_false_if_the_building_has_no_remaining_capacity_of_the_resource(self): building = Building() building.container.add_resource_slot(Fish, 10) building.container.add_resource_slot(Cabbage, 10) building.container.add_resource_slot(Stone, 10) building.receive_cargo(Fish, 10) cabbage_plant = ResourcePlant() cabbage_plant.add_resource_requirement(Fish, 3) cabbage_plant.add_resource_product(Cabbage, 2) building.add_resource_plant(cabbage_plant) produce = Produce(building, Cabbage, 1) self.assertFalse(produce.is_possible(building, 1))
def test__execute_produces_resource_based_on_the_transfer_rate_and_dt(self): building = Building() building.container.add_resource_slot(Fish, 10) building.container.add_resource_slot(Cabbage, 10) building.container.add_resource_slot(Stone, 10) building.receive_cargo(Fish, 10) cabbage_plant = ResourcePlant() cabbage_plant.add_resource_requirement(Fish, 3) cabbage_plant.add_resource_product(Cabbage, 2) building.add_resource_plant(cabbage_plant) produce = Produce(building, Cabbage, 1) self.assertTrue(produce.is_possible(building, 1)) produce._execute(building, 2) self.assertEqual(building.container.current_load(Cabbage), 2)
def test_digest_consumes_required_resources_and_produces_resource_products(self): # noqa tray = SlottedCargoContainer() tray.add_resource_slot(Fish, 10) tray.add_resource_slot(Wood, 10) tray.add_resource_slot(Cabbage, 10) tray.load_cargo(Fish, 3) tray.load_cargo(Wood, 5) tray.load_cargo(Cabbage, 4) plant = ResourcePlant() plant.add_resource_requirement(Fish, 3) plant.add_resource_requirement(Wood, 2) plant.add_resource_product(Cabbage, 6) plant.digest(tray) self.assertEqual(tray.cargo_slots[Fish.name]['load'], 0) self.assertEqual(tray.cargo_slots[Wood.name]['load'], 3) self.assertEqual(tray.cargo_slots[Cabbage.name]['load'], 10)
def test_produce_resources_digests_resources_with_all_the_available_digesters(self): # noqa building = Building() building.container.add_resource_slot(Fish, 10) building.container.add_resource_slot(Wood, 10) building.container.add_resource_slot(Cabbage, 10) building.container.add_resource_slot(Stone, 10) building.receive_cargo(Fish, 10) building.receive_cargo(Wood, 10) cabbage_plant = ResourcePlant() cabbage_plant.add_resource_requirement(Fish, 3) cabbage_plant.add_resource_requirement(Wood, 2) cabbage_plant.add_resource_product(Cabbage, 2) building.add_resource_plant(cabbage_plant) stone_plant = ResourcePlant() stone_plant.add_resource_requirement(Fish, 1) stone_plant.add_resource_requirement(Wood, 3) stone_plant.add_resource_requirement(Cabbage, 5) stone_plant.add_resource_product(Stone, 1) building.add_resource_plant(stone_plant) building.produce_resources() self.assertEqual(building.container.current_load(Fish), 7) self.assertEqual(building.container.current_load(Wood), 8) self.assertEqual(building.container.current_load(Cabbage), 2) self.assertEqual(building.container.current_load(Stone), 0) building.produce_resources() self.assertEqual(building.container.current_load(Fish), 4) self.assertEqual(building.container.current_load(Wood), 6) self.assertEqual(building.container.current_load(Cabbage), 4) self.assertEqual(building.container.current_load(Stone), 0) building.produce_resources() self.assertEqual(building.container.current_load(Fish), 0) self.assertEqual(building.container.current_load(Wood), 1) self.assertEqual(building.container.current_load(Cabbage), 1) self.assertEqual(building.container.current_load(Stone), 1)