Functor/3 fix

tuProlog · Jun 16, 2020 · e52f64b · e52f64b
1 parent 53d98ff
commit e52f64b
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/solve/src/commonMain/kotlin/it/unibo/tuprolog/solve/stdlib/primitive/Functor.kt b/solve/src/commonMain/kotlin/it/unibo/tuprolog/solve/stdlib/primitive/Functor.kt
@@ -30,7 +30,10 @@ object Functor : TernaryRelation.Functional<ExecutionContext>("functor") {
                                 Substitution.failed()
                         }
                         is Var -> {
-                            Substitution.of(third to Integer.of(first.arity))
+                            if (first.functor == second.value)
+                                Substitution.of(third to Integer.of(first.arity))
+                            else
+                                Substitution.failed()
                         }
                         else -> {
                             // TODO expected here should be INTEGER | VARIABLE
@@ -41,7 +44,10 @@ object Functor : TernaryRelation.Functional<ExecutionContext>("functor") {
                 is Var -> {
                     when (third) {
                         is Numeric -> {
-                            Substitution.of(second to Atom.of(first.functor))
+                            if (first.arity == third.intValue.toInt())
+                                Substitution.of(second to Atom.of(first.functor))
+                            else
+                                Substitution.failed()
                         }
                         is Var -> {
                             Substitution.of(