Merge pull request #135 from AI-Planning/extra-constants-tests

Extra constants tests

Author: haz

pddl/__version__.py

@@ -15,7 +15,7 @@
 __title__ = "pddl"
 __description__ = "PDDL parser"
 __url__ = "https://github.com/AI-Planning/pddl.git"
-__version__ = "0.4.2"
+__version__ = "0.4.3"
 __author__ = "Marco Favorito, Francesco Fuggitti, Christian Muise"
 __author_email__ = (
     "marco.favorito@gmail.com, francesco.fuggitti@gmail.com, christian.muise@queensu.ca"

tests/test_functions.py

@@ -14,7 +14,7 @@
 import pytest
 
 from pddl.logic.functions import Metric, NumericFunction, NumericValue
-from pddl.logic.helpers import variables
+from pddl.logic.helpers import constants, variables
 from pddl.parser.symbols import Symbols
 
 
@@ -24,7 +24,9 @@ class TestNumericFunction:
     def setup_method(self):
         """Set up the tests."""
         self.a, self.b = variables("a b", types=["type1"])
+        self.c = constants("c")[0]
         self.function = NumericFunction("func", self.a, self.b)
+        self.function2 = NumericFunction("func2", self.a, self.c)
 
     def test_name(self):
         """Test name getter."""
@@ -46,6 +48,7 @@ def test_to_equal(self):
     def test_to_str(self):
         """Test to string."""
         assert str(self.function) == f"({self.function.name} {self.a} {self.b})"
+        assert str(self.function2) == f"({self.function2.name} {self.a} {self.c})"
 
     def test_to_repr(self):
         """Test to repr."""

tests/test_predicate.py

@@ -13,7 +13,7 @@
 """This module contains tests for PDDL predicates."""
 from pddl.core import Predicate
 from pddl.logic.base import And
-from pddl.logic.helpers import variables
+from pddl.logic.helpers import constants, variables
 from pddl.logic.predicates import DerivedPredicate, EqualTo
 
 
@@ -23,7 +23,9 @@ class TestPredicateSimpleInitialisation:
     def setup_method(self):
         """Set up the tests."""
         self.a, self.b = variables("a b")
+        self.c = constants("c")[0]
         self.predicate = Predicate("P", self.a, self.b)
+        self.predicate2 = Predicate("P2", self.a, self.c)
 
     def test_name(self):
         """Test name getter."""
@@ -45,6 +47,7 @@ def test_to_equal(self):
     def test_to_str(self):
         """Test to string."""
         assert str(self.predicate) == f"({self.predicate.name} {self.a} {self.b})"
+        assert str(self.predicate2) == f"({self.predicate2.name} {self.a} {self.c})"
 
     def test_to_repr(self):
         """Test to repr."""