test: landscape module

tests/conftest.py

@@ -14,3 +14,10 @@ def pygame():
     yield
 
     pg.quit()
+
+
+@pytest.fixture
+def pg_screen_mock(mocker):
+    """Mock the pygame screen surface."""
+
+    return mocker.Mock(spec=pg.Surface, instance=True)

tests/test_landscape.py

@@ -0,0 +1,22 @@
+from __future__ import annotations
+
+from doggo.landscape import Biome
+from doggo.landscape import LandscapeLayer
+
+
+def test_get_random_biome():
+    biome = Biome.random()
+
+    assert isinstance(biome, Biome)
+    assert biome in list(Biome)
+
+
+def test_landscape_layer_draw_on_screen(pg_screen_mock):
+    image = f"{Biome.MOUNTAIN:02d}_bg.png"
+    landscape_layer = LandscapeLayer(image=image)
+
+    landscape_layer.draw(pg_screen_mock)
+
+    pg_screen_mock.blit.assert_called_once_with(
+        landscape_layer.image, landscape_layer.rect
+    )