(agrf) More test cases

ahyangyi · Nov 15, 2024 · 0e1d502 · 0e1d502
1 parent bcb2a41
commit 0e1d502
Showing 1 changed file with 18 additions and 1 deletion.
diff --git 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