✅ Fix optional dependencies in tests

tests/examples/minecraft/test_behaviors.py

@@ -3,7 +3,6 @@
 from pathlib import Path
 from typing import TYPE_CHECKING
 
-import matplotlib.pyplot as plt
 import pytest
 import pytest_check as check
 
@@ -26,6 +25,7 @@
 
 if TYPE_CHECKING:
     from unified_planning.io import PDDLWriter
+    import matplotlib.pyplot
 
 WOODEN_PICKAXE = MC_TOOLS_BY_TYPE_AND_MATERIAL[ToolType.PICKAXE][Material.WOOD]
 STONE_PICKAXE = MC_TOOLS_BY_TYPE_AND_MATERIAL[ToolType.PICKAXE][Material.STONE]
@@ -158,6 +158,7 @@ def test_graph_rolled(self):
         )
 
     def test_graph_draw(self):
+        plt: "matplotlib.pyplot" = pytest.importorskip("matplotlib.pyplot")
         fig, ax = plt.subplots()
         self.pickaxe_behavior.graph.draw(ax)
         plt.close(fig)

tests/examples/test_common.py

@@ -1,5 +1,5 @@
 from pathlib import Path
-from typing import Type
+from typing import TYPE_CHECKING, Type
 
 import pytest
 import pytest_check as check
@@ -17,7 +17,9 @@
     MiniHCraftUnlockPickup,
 )
 from hcraft.env import HcraftEnv
-from hcraft.requirements import apply_color_theme
+
+if TYPE_CHECKING:
+    import matplotlib.pyplot
 
 
 @pytest.mark.slow
@@ -108,7 +110,7 @@ def test_requirements_graph(env_class: Type[HcraftEnv], mocker: MockerFixture):
     requirements_dir = Path("docs", "images", "requirements_graphs")
 
     if draw_plt:
-        import matplotlib.pyplot as plt
+        plt: "matplotlib.pyplot" = pytest.importorskip("matplotlib.pyplot")
 
         width = max(requirements.depth, 10)
         height = max(9 / 16 * width, requirements.width / requirements.depth * width)
@@ -127,6 +129,7 @@ def test_requirements_graph(env_class: Type[HcraftEnv], mocker: MockerFixture):
         plt.close()
 
     if draw_html:
+        pytest.importorskip("pyvis")
         mocker.patch("pyvis.network.webbrowser.open")
         requirements_dir.mkdir(exist_ok=True)
         filepath = requirements_dir / f"{env.name}_requirements_graph.html"