Batch crop farming and stop on unreachable barrels
This commit is contained in:
@@ -0,0 +1,35 @@
|
||||
import unittest
|
||||
from munch import Munch
|
||||
|
||||
from mosfet.info import blocks
|
||||
from mosfet.jobs.gather_crop import GatherCropStates
|
||||
|
||||
|
||||
class CropBatchTests(unittest.TestCase):
|
||||
def test_gather_crop_finishes_after_sixteen_successful_replants(self):
|
||||
unripe_wheat = blocks.MCD_BLOCKS['wheat']['minStateId']
|
||||
mature_wheat = blocks.MATURE_WHEAT_ID
|
||||
self.assertNotEqual(unripe_wheat, mature_wheat)
|
||||
|
||||
class FakeWorld:
|
||||
@staticmethod
|
||||
def block_at(*_position):
|
||||
return unripe_wheat
|
||||
|
||||
state = GatherCropStates(Munch(world=FakeWorld()))
|
||||
state.crop = (1, 64, 1)
|
||||
state.type_id = mature_wheat
|
||||
|
||||
for completed in range(1, state.CROPS_PER_BATCH):
|
||||
state.wait_place()
|
||||
self.assertEqual(state.completed_crops, completed)
|
||||
self.assertIs(state.state.__func__, state.find_new_crop.__func__)
|
||||
state.state = state.wait_place
|
||||
|
||||
state.wait_place()
|
||||
self.assertEqual(state.completed_crops, 0)
|
||||
self.assertIs(state.state.__func__, state.cleanup.__func__)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.main()
|
||||
@@ -44,6 +44,20 @@ class GroundSupplyTests(unittest.TestCase):
|
||||
self.assertEqual(state.target_items, (items.CHEST_ID,))
|
||||
self.assertIs(state.state.__func__, state.find_ground_items.__func__)
|
||||
|
||||
def test_unreachable_barrel_ends_supply_task_without_trying_more(self):
|
||||
class FakeWorld:
|
||||
@staticmethod
|
||||
def path_to_place_faked(_start, _barrel):
|
||||
return None
|
||||
|
||||
state = GrabSuppliesStates(Munch(
|
||||
pos=(0.5, 64.0, 0.5), world=FakeWorld(), chunks=Munch()))
|
||||
state.barrel = (10, 64, 10)
|
||||
state.path_to_barrel()
|
||||
|
||||
self.assertIs(state.state.__func__, state.cleanup.__func__)
|
||||
self.assertEqual(state.bad_barrels, [(10, 64, 10)])
|
||||
|
||||
def test_job_walks_to_and_picks_up_dropped_supply_before_barrels(self):
|
||||
axe_id = next(iter(items.AXE_IDS))
|
||||
dropped_axe = Munch(
|
||||
|
||||
Reference in New Issue
Block a user