diff --git a/agrf/lib/building/layout_test.py b/agrf/lib/building/layout_test.py index 6937cc16..d02ef287 100644 --- a/agrf/lib/building/layout_test.py +++ b/agrf/lib/building/layout_test.py @@ -1,4 +1,4 @@ -from agrf.lib.building.layout import ADefaultGroundSprite +from agrf.lib.building.layout import ADefaultGroundSprite, ALayout def test_default_groundsprite(): @@ -7,3 +7,20 @@ def test_default_groundsprite(): graphics = gs1012.graphics(4, 32).to_image() assert graphics.shape == (127, 256, 4) assert graphics[64, 128, 1] == 67 + + m_graphics = gs1012.M.graphics(4, 32).to_image() + assert m_graphics.shape == (127, 256, 4) + assert m_graphics[64, 128, 1] == 72 + + +def test_layout(): + gs1012 = ADefaultGroundSprite(1012) + l = ALayout(gs1012, [], True) + + graphics = l.graphics(4, 32).to_image() + assert graphics.shape == (127, 256, 4) + assert graphics[64, 128, 1] == 67 + + m_graphics = l.M.graphics(4, 32).to_image() + assert m_graphics.shape == (127, 256, 4) + assert m_graphics[64, 128, 1] == 72