def test_single_1x2(self):
bp = Blueprint(**BlueprintDict(entities=[
get_entity_dict("splitter", 1, 0.5),
]))
g = Grid(bp)
self.assertEqual(2, g.width)
self.assertEqual(1, g.height)
self.assertEqual(bp.entities[0], g[0, 0])
self.assertEqual(bp.entities[0], g[1, 0])
bp = Blueprint(**BlueprintDict(entities=[
get_entity_dict("splitter", 0.5, 1, direction=2),
]))
g = Grid(bp)
self.assertEqual(1, g.width)
self.assertEqual(2, g.height)
self.assertEqual(bp.entities[0], g[0, 0])
self.assertEqual(bp.entities[0], g[0, 1])
bp = Blueprint(**BlueprintDict(entities=[
get_entity_dict("splitter", -1, 0.5),
]))
g = Grid(bp)
self.assertEqual(2, g.width)
self.assertEqual(1, g.height)
self.assertEqual(bp.entities[0], g[0, 0])
self.assertEqual(bp.entities[0], g[1, 0])
def test_multiple_1x1(self):
bp = Blueprint(**BlueprintDict(entities=[
get_entity_dict("transport-belt", 0.5, 0.5),
get_entity_dict("transport-belt", 1.5, 1.5),
]))
g = Grid(bp)
self.assertEqual(2, g.width)
self.assertEqual(2, g.height)
self.assertEqual(bp.entities[0], g[0, 0])
self.assertEqual(bp.entities[1], g[1, 1])
self.assertIsNone(g[0, 1])
self.assertIsNone(g[1, 0])
bp = Blueprint(**BlueprintDict(entities=[
get_entity_dict("transport-belt", -2.5, -2.5),
get_entity_dict("transport-belt", -2.5, 2.5),
get_entity_dict("transport-belt", 2.5, -2.5),
get_entity_dict("transport-belt", 2.5, 2.5),
]))
g = Grid(bp)
self.assertEqual(6, g.width)
self.assertEqual(6, g.height)
self.assertEqual(bp.entities[0], g[0, 0])
self.assertEqual(bp.entities[1], g[0, 5])
self.assertEqual(bp.entities[2], g[5, 0])
self.assertEqual(bp.entities[3], g[5, 5])
def test_single_1x1(self):
bp = Blueprint(**BlueprintDict(entities=[
get_entity_dict("transport-belt", 0.5, 0.5),
]))
g = Grid(bp)
self.assertEqual(1, g.width)
self.assertEqual(1, g.height)
self.assertEqual(bp.entities[0], g[0, 0])
bp = Blueprint(**BlueprintDict(entities=[
get_entity_dict("transport-belt", 1.5, 0.5),
]))
g = Grid(bp)
self.assertEqual(1, g.width)
self.assertEqual(1, g.height)
self.assertEqual(bp.entities[0], g[0, 0])
bp = Blueprint(**BlueprintDict(entities=[
get_entity_dict("transport-belt", -0.5, 0.5),
]))
g = Grid(bp)
self.assertEqual(1, g.width)
self.assertEqual(1, g.height)
self.assertEqual(bp.entities[0], g[0, 0])
bp = Blueprint(**BlueprintDict(entities=[
get_entity_dict("transport-belt", -1.5, 0.5),
]))
g = Grid(bp)
self.assertEqual(1, g.width)
self.assertEqual(1, g.height)
self.assertEqual(bp.entities[0], g[0, 0])
def test_bounding_box_single_3x3(self):
bp = Blueprint(**BlueprintDict(entities=[get_entity_dict("assembling-machine-1", 0.5, 0.5)]))
self.assertEqual((Position(-1, -1), Position(2, 2)), bp.bounding_box)
bp = Blueprint(**BlueprintDict(entities=[get_entity_dict("assembling-machine-1", 1.5, 1.5)]))
self.assertEqual((Position(0, 0), Position(3, 3)), bp.bounding_box)
bp = Blueprint(**BlueprintDict(entities=[get_entity_dict("assembling-machine-1", -1.5, -1.5)]))
self.assertEqual((Position(-3, -3), Position(0, 0)), bp.bounding_box)
def __init__(self,
blueprint: Blueprint,
width=None,
height=None,
x_offset=None,
y_offset=None):
self.blueprint = blueprint
low, high = blueprint.bounding_box
if width is None or height is None:
self.width, self.height = blueprint.get_dimensions()
else:
self.width = width
self.height = height
if x_offset is None or y_offset is None:
self.x_offset = np.floor(low.x)
self.y_offset = np.floor(low.y)
else:
self.x_offset = x_offset
self.y_offset = y_offset
self.grid = np.empty(shape=(self.height, self.width), dtype=Entity)
self._yd, self._xd = np.indices((self.height, self.width))
self._yd = self._yd + self.y_offset
self._xd = self._xd + self.x_offset
for e in blueprint.entities:
self.add_entity(e)
def test_mixed(self):
bp = Blueprint(**BlueprintDict(entities=[
get_entity_dict("assembling-machine-1", 9.5, -5.5),
get_entity_dict("electric-furnace", 6.5, -3.5),
get_entity_dict("transport-belt", 8.5, -3.5),
get_entity_dict("splitter", 5.5, -7, direction=2),
]))
g = Grid(bp)
self.assertEqual(6, g.width)
self.assertEqual(6, g.height)
self.assertEqual(g[0, 0], bp.entities[3])
self.assertEqual(g[0, 1], bp.entities[3])
self.assertEqual(g[3, 1], bp.entities[0])
self.assertEqual(g[4, 1], bp.entities[0])
self.assertEqual(g[5, 1], bp.entities[0])
self.assertEqual(g[3, 2], bp.entities[0])
self.assertEqual(g[4, 2], bp.entities[0])
self.assertEqual(g[5, 2], bp.entities[0])
self.assertEqual(g[3, 3], bp.entities[0])
self.assertEqual(g[4, 3], bp.entities[0])
self.assertEqual(g[5, 3], bp.entities[0])
self.assertEqual(g[0, 3], bp.entities[1])
self.assertEqual(g[1, 3], bp.entities[1])
self.assertEqual(g[2, 3], bp.entities[1])
self.assertEqual(g[0, 4], bp.entities[1])
self.assertEqual(g[1, 4], bp.entities[1])
self.assertEqual(g[2, 4], bp.entities[1])
self.assertEqual(g[0, 5], bp.entities[1])
self.assertEqual(g[1, 5], bp.entities[1])
self.assertEqual(g[2, 5], bp.entities[1])
self.assertEqual(g[3, 4], bp.entities[2])
def test_get_bounding_box_3x3_square(self):
bp = Blueprint(**BlueprintDict(entities=[
get_entity_dict("transport-belt", 0.5, 0.5),
get_entity_dict("transport-belt", 2.5, 2.5),
]))
self.assertEqual((Position(0, 0), Position(3, 3)), bp.bounding_box)
bp = Blueprint(**BlueprintDict(entities=[
get_entity_dict("transport-belt", -0.5, -0.5),
get_entity_dict("transport-belt", 1.5, 1.5),
]))
self.assertEqual((Position(-1, -1), Position(2, 2)), bp.bounding_box)
bp = Blueprint(**BlueprintDict(entities=[
get_entity_dict("transport-belt", -2.5, -2.5),
get_entity_dict("transport-belt", -0.5, -0.5),
]))
self.assertEqual((Position(-3, -3), Position(0, 0)), bp.bounding_box)
def test_get_dimensions_single_rails(self):
bp = Blueprint(**BlueprintDict(entities=[
get_entity_dict("curved-rail", 0, 0)
]))
self.assertEqual((5, 8), bp.get_dimensions())
bp = Blueprint(**BlueprintDict(entities=[
get_entity_dict("curved-rail", 0, 0, direction=5)
]))
self.assertEqual((5, 8), bp.get_dimensions())
bp = Blueprint(**BlueprintDict(entities=[
get_entity_dict("curved-rail", 0, 0, direction=2)
]))
self.assertEqual((8, 5), bp.get_dimensions())
def _main():
bp = Blueprint(**BlueprintDict(entities=[
get_entity_dict("transport-belt", -3.5, 0.5),
get_entity_dict("transport-belt", 2.5, 2.5),
get_entity_dict("assembling-machine-1", 5.5, 2.5),
get_entity_dict("splitter", 0.5, 2, direction=2),
]))
print(bp)
g: Grid = Grid(bp)
print(g)
def test_multiple_1x2(self):
bp = Blueprint(**BlueprintDict(entities=[
get_entity_dict("splitter", 1, 0.5),
get_entity_dict("splitter", 1.5, 2, direction=2),
get_entity_dict("splitter", -0.5, 2),
]))
g = Grid(bp)
self.assertIsNone(g[0, 0])
self.assertEqual(bp.entities[0], g[1, 0])
self.assertEqual(bp.entities[0], g[2, 0])
self.assertIsNone(g[0, 1])
self.assertIsNone(g[1, 1])
self.assertEqual(bp.entities[1], g[2, 1])
self.assertEqual(bp.entities[2], g[0, 2])
self.assertEqual(bp.entities[2], g[1, 2])
self.assertEqual(bp.entities[1], g[2, 2])
def __init__(self,
blueprint: BlueprintDict = None,
blueprint_book: BlueprintBookDict = None,
upgrade_planner: UpgradePlannerDict = None,
deconstruction_planner: DeconstructionPlannerDict = None,
*args,
**kwargs):
# Delayed import to prevent import errors
global BlueprintBookDict, BlueprintBook, BlueprintDict, Blueprint
from Blueprint.BlueprintBook import BlueprintBookDict, BlueprintBook
from Blueprint.Blueprint import BlueprintDict, Blueprint
self.blueprint_book: BlueprintBook = None if blueprint_book is None else BlueprintBook(
**blueprint_book)
self.blueprint: Blueprint = None if blueprint is None else Blueprint(
**blueprint)
def test_get_dimensions_3x3_square(self):
bp = Blueprint(**BlueprintDict(entities=[
get_entity_dict("transport-belt", 0.5, 0.5),
get_entity_dict("transport-belt", 2.5, 2.5),
]))
self.assertEqual((3, 3), bp.get_dimensions())
bp = Blueprint(**BlueprintDict(entities=[
get_entity_dict("assembling-machine-1", -0.5, -0.5),
get_entity_dict("transport-belt", 2.5, 2.5),
]))
self.assertEqual((5, 5), bp.get_dimensions())
def test_get_dimensions_single_3x3(self):
bp = Blueprint(**BlueprintDict(entities=[get_entity_dict("assembling-machine-1", 1.5, 1.5)]))
self.assertEqual((3, 3), bp.get_dimensions())
def test_bounding_box_single_1x1(self):
bp = Blueprint(**BlueprintDict(entities=[get_entity_dict("transport-belt", 0.5, 0.5)]))
self.assertEqual((Position(0, 0), Position(1, 1)), bp.bounding_box)
bp = Blueprint(**BlueprintDict(entities=[get_entity_dict("transport-belt", -0.5, -0.5)]))
self.assertEqual((Position(-1, -1), Position(0, 0)), bp.bounding_box)
def test_get_dimensions_mixed(self):
bp = Blueprint(**BlueprintDict(entities=[
get_entity_dict("stone-furnace", -10, 0),
get_entity_dict("stone-furnace", 10, 0),
]))
self.assertEqual((22, 2), bp.get_dimensions())
def test_get_dimensions_multiple_rails(self):
bp = Blueprint(**BlueprintDict(entities=[
get_entity_dict("curved-rail", 0, 0),
get_entity_dict("curved-rail", 18, 0)
]))
self.assertEqual((23, 8), bp.get_dimensions())
bp = Blueprint(**BlueprintDict(entities=[
get_entity_dict("curved-rail", 0, 0, direction=0),
get_entity_dict("curved-rail", 18, 0, direction=4)
]))
self.assertEqual((24, 8), bp.get_dimensions())
bp = Blueprint(**BlueprintDict(entities=[
get_entity_dict("curved-rail", 0, 0, direction=2),
get_entity_dict("curved-rail", 0, 18, direction=2)
]))
self.assertEqual((8, 23), bp.get_dimensions())
bp = Blueprint(**BlueprintDict(entities=[
get_entity_dict("curved-rail", 0, 0, direction=2),
get_entity_dict("curved-rail", 0, 18, direction=6)
]))
self.assertEqual((8, 24), bp.get_dimensions())
def test_get_dimensions_single_1x1(self):
bp = Blueprint(**BlueprintDict(entities=[get_entity_dict("transport-belt", 0.5, 0.5)]))
self.assertEqual((1, 1), bp.get_dimensions())