Add tests for atomic skeletons

sunsided · May 19, 2024 · 0f44b2a · 0f44b2a
1 parent 1b19e59
commit 0f44b2a
Show file tree

Hide file tree

Showing 2 changed files with 41 additions and 2 deletions.
diff --git a/src/parsers/atomic_formula_skeleton.rs b/src/parsers/atomic_formula_skeleton.rs
@@ -59,3 +59,25 @@ impl crate::parsers::Parser for AtomicFormulaSkeleton {
         parse_atomic_formula_skeleton(input)
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use crate::{AtomicFormulaSkeleton, Parser, Predicate, Variable};
+    use crate::{ToTyped, TypedList};
+
+    #[test]
+    fn test_parse() {
+        let (_, value) = AtomicFormulaSkeleton::parse("(at ?x - physob ?y - location)").unwrap();
+
+        assert_eq!(
+            value,
+            AtomicFormulaSkeleton::new(
+                Predicate::from("at"),
+                TypedList::from_iter([
+                    Variable::from("x").to_typed("physob"),
+                    Variable::from("y").to_typed("location")
+                ])
+            )
+        );
+    }
+}
diff --git a/src/parsers/atomic_function_skeleton.rs b/src/parsers/atomic_function_skeleton.rs
@@ -40,8 +40,7 @@ impl crate::parsers::Parser for AtomicFunctionSkeleton {
     ///
     /// ## Example
     /// ```
-    /// # use pddl::parsers::{parse_atomic_function_skeleton, Span, UnwrapValue};
-    /// # use pddl::{Variable, AtomicFunctionSkeleton, Predicate, FunctionSymbol, Parser};
+    /// # use pddl::{Variable, AtomicFunctionSkeleton, FunctionSymbol, Parser};
     /// # use pddl::{ToTyped, TypedList};
     /// let (_, value) = AtomicFunctionSkeleton::parse("(battery-amount ?r - rover)").unwrap();
     ///
@@ -60,3 +59,21 @@ impl crate::parsers::Parser for AtomicFunctionSkeleton {
         parse_atomic_function_skeleton(input)
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use crate::{AtomicFunctionSkeleton, FunctionSymbol, Parser, Variable};
+    use crate::{ToTyped, TypedList};
+
+    #[test]
+    fn test_parse() {
+        let (_, value) = AtomicFunctionSkeleton::parse("(battery-amount ?r - rover)").unwrap();
+        assert_eq!(
+            value,
+            AtomicFunctionSkeleton::new(
+                FunctionSymbol::from("battery-amount"),
+                TypedList::from_iter([Variable::from("r").to_typed("rover")])
+            )
+        );
+    }
+}