(station/tests) Refactor some test cases to use whole-image comparison

Author: ahyangyi

agrf/lib/building/layout_test.py

@@ -1,28 +1,32 @@
 import grf
+import numpy as np
+from PIL import Image
 from agrf.lib.building.layout import AGroundSprite, ADefaultGroundSprite, ALayout
 from agrf.lib.building.symmetry import BuildingSymmetrical
 from agrf.lib.building.image_sprite import image_sprite
 from agrf.graphics.misc import SCALE_TO_ZOOM
 
+temperate_1011 = np.array(Image.open("agrf/third_party/opengfx2/temperate/1011.png"))
+temperate_1012 = np.array(Image.open("agrf/third_party/opengfx2/temperate/1012.png"))
+
+
+gs1012 = ADefaultGroundSprite(1012)
+
 
 def test_default_groundsprite():
-    gs1012 = ADefaultGroundSprite(1012)
+    assert (temperate_1012 == gs1012.graphics(4, 32).to_image()).all()
 
-    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_default_groundsprite_M():
+    assert (temperate_1011 == gs1012.M.graphics(4, 32).to_image()).all()
 
-    r_graphics = gs1012.R.graphics(4, 32).to_image()
-    assert r_graphics.shape == (127, 256, 4)
-    assert r_graphics[64, 128, 1] == 67
 
-    t_graphics = gs1012.T.graphics(4, 32).to_image()
-    assert t_graphics.shape == (127, 256, 4)
-    assert t_graphics[64, 128, 1] == 67
+def test_default_groundsprite_R():
+    assert (temperate_1012 == gs1012.R.graphics(4, 32).to_image()).all()
+
+
+def test_default_groundsprite_T():
+    assert (temperate_1012 == gs1012.T.graphics(4, 32).to_image()).all()
 
 
 def test_groundsprite():
@@ -33,7 +37,6 @@ def test_groundsprite():
 
 
 def test_layout():
-    gs1012 = ADefaultGroundSprite(1012)
     l = ALayout(gs1012, [], True)
 
     graphics = l.graphics(4, 32).to_image()