Apply sign work zones before candidate routing
This commit is contained in:
@@ -0,0 +1,68 @@
|
||||
import unittest
|
||||
from munch import Munch
|
||||
|
||||
from mosfet.info import blocks
|
||||
from mosfet.jobs.gather_crop import GatherCropStates
|
||||
from mosfet.jobs.gather_wood import GatherWoodStates
|
||||
|
||||
|
||||
class SignWorkTargetTests(unittest.TestCase):
|
||||
def test_crop_search_filters_out_of_bounds_candidates_before_navigation(self):
|
||||
outside = (10, 64, 0)
|
||||
mature_wheat = blocks.MATURE_WHEAT_ID
|
||||
crop_state = blocks.MCD_BLOCKS['wheat']['minStateId']
|
||||
|
||||
class FakeWorld:
|
||||
def find_blocks_indexed(self, _center, _block_ids, _distance):
|
||||
return [outside]
|
||||
|
||||
@staticmethod
|
||||
def work_position_allowed(position):
|
||||
return position != outside
|
||||
|
||||
def filter_work_positions(self, positions):
|
||||
return [position for position in positions
|
||||
if self.work_position_allowed(position)]
|
||||
|
||||
@staticmethod
|
||||
def block_at(*_position):
|
||||
return crop_state
|
||||
|
||||
state = GatherCropStates(Munch(world=FakeWorld(), pos=(0, 64, 0)))
|
||||
state.find_new_crop()
|
||||
|
||||
self.assertIs(state.state.__func__, state.wait_for_crops.__func__)
|
||||
self.assertEqual(state.bad_crops, [])
|
||||
|
||||
def test_tree_search_skips_out_of_bounds_tree_without_blacklisting_it(self):
|
||||
outside = (10, 64, 0)
|
||||
inside = (2, 64, 0)
|
||||
oak_log = blocks.MCD_BLOCKS['oak_log']['minStateId']
|
||||
|
||||
class FakeWorld:
|
||||
@staticmethod
|
||||
def find_trees(_center, _distance):
|
||||
return [outside, inside]
|
||||
|
||||
@staticmethod
|
||||
def work_position_allowed(position):
|
||||
return position != outside
|
||||
|
||||
def filter_work_positions(self, positions):
|
||||
return [position for position in positions
|
||||
if self.work_position_allowed(position)]
|
||||
|
||||
@staticmethod
|
||||
def block_at(*_position):
|
||||
return oak_log
|
||||
|
||||
state = GatherWoodStates(Munch(world=FakeWorld(), pos=(0, 64, 0)))
|
||||
state.find_new_tree()
|
||||
|
||||
self.assertEqual(state.tree, inside)
|
||||
self.assertEqual(state.type, 'oak')
|
||||
self.assertEqual(state.bad_trees, [])
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.main()
|
||||
Reference in New Issue
Block a user