Fix bug in the new test

Farama-Foundation · Jun 19, 2023 · 89c2a05 · 89c2a05
1 parent e8d5817
commit 89c2a05
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/tests/test_docstrings.py b/tests/test_docstrings.py
@@ -3,7 +3,7 @@
 import vizdoom as vzd
 
 
-def test_docstrings(object):
+def _check_object_docstrings(object):
     object_methods = [
         method_name
         for method_name in dir(object)
@@ -14,5 +14,9 @@ def test_docstrings(object):
         assert method.__doc__ is not None, f"Method {method} has no docstring"
 
 
+def test_docstrings():
+    _check_object_docstrings(vzd.DoomGame)
+
+
 if __name__ == "__main__":
-    test_docstrings(vzd.DoomGame)
+    test_docstrings()