add more tests

diff --git a/src/main.rs b/src/main.rs
index 962c35d096d74c34da5b0fa89c8780d2fb6132de..56c9b0488002c97b560302f87998bd50ad0d9e32 100644 (file)
--- a/src/main.rs
+++ b/src/main.rs
@@ -625,6 +625,19 @@ mod tests {
         submit(&mut wam, "p(f(g(X)), compound, [lists,are,good]).");
 
         assert_eq!(submit(&mut wam, "?- call(p(f(g(X))), Y, Z)."), true);
+
+        submit(&mut wam, "david_lynch(coffee).
+                          david_lynch(pie).
+                          david_lynch(kyle(Film)) :- kyle(Film).");
+
+        submit(&mut wam, "kyle(dune).
+                          kyle(blue_velvet).
+                          kyle(showgirls).
+                          kyle(flintstones).");
+
+        assert_eq!(submit(&mut wam, "?- call(david_lynch, X)."), true);
+        assert_eq!(submit(&mut wam, "?- call(david_lynch, kyle(Film))."), true);
+        assert_eq!(submit(&mut wam, "?- call(david_lynch, kyle(Film), _)."), false);
     }
 }