Batch crop farming and stop on unreachable barrels

This commit is contained in:
Tanner
2026-11-05 23:37:10 +00:00
parent ef7bf1d7da
commit 09035793e2
5 changed files with 63 additions and 7 deletions
+35
View File
@@ -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()
+14
View File
@@ -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(