Checkpoint
This commit is contained in:
parent
43b364044e
commit
9db1c66781
34 changed files with 746 additions and 194 deletions
|
@ -2,6 +2,7 @@ package prolog
|
|||
|
||||
import org.junit.jupiter.api.Assertions.*
|
||||
import org.junit.jupiter.api.BeforeEach
|
||||
import org.junit.jupiter.api.Nested
|
||||
import org.junit.jupiter.api.Test
|
||||
import prolog.ast.logic.Fact
|
||||
import prolog.ast.logic.Rule
|
||||
|
@ -108,9 +109,9 @@ class EvaluationTests {
|
|||
val parent = Rule(
|
||||
Structure(Atom("parent"), listOf(variable1, variable2)),
|
||||
/* :- */ Disjunction(
|
||||
Structure(Atom("father"), listOf(variable1, variable2)),
|
||||
/* ; */
|
||||
Structure(Atom("mother"), listOf(variable1, variable2))
|
||||
Structure(Atom("father"), listOf(variable1, variable2)),
|
||||
/* ; */
|
||||
Structure(Atom("mother"), listOf(variable1, variable2))
|
||||
)
|
||||
)
|
||||
|
||||
|
@ -212,7 +213,182 @@ class EvaluationTests {
|
|||
assertEquals(expectedResults.size, actualResults.size, "Number of results should match")
|
||||
for (i in expectedResults.indices) {
|
||||
assertEquals(expectedResults[i].size, actualResults[i].getOrNull()!!.size, "Substitution size should match")
|
||||
assertTrue(expectedResults[i].all { actualResults[i].getOrNull()!![it.key]?.let { it1 -> equivalent(it.value, it1, emptyMap()) } ?: false }, "Substitution values should match")
|
||||
assertTrue(expectedResults[i].all {
|
||||
actualResults[i].getOrNull()!![it.key]?.let { it1 ->
|
||||
equivalent(
|
||||
it.value,
|
||||
it1,
|
||||
emptyMap()
|
||||
)
|
||||
} ?: false
|
||||
}, "Substitution values should match")
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `likes(alice, pizza)`() {
|
||||
val fact = Fact(Structure(Atom("likes"), listOf(Atom("alice"), Atom("pizza"))))
|
||||
val goal = Structure(Atom("likes"), listOf(Atom("alice"), Atom("pizza")))
|
||||
|
||||
Program.load(listOf(fact))
|
||||
|
||||
val result = Program.query(goal).toList()
|
||||
|
||||
assertEquals(1, result.size, "Expected 1 result")
|
||||
assertTrue(result[0].isSuccess, "Expected success")
|
||||
val subs = result[0].getOrNull()!!
|
||||
assertEquals(0, subs.size, "Expected no substitutions")
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `likes(Person, pizza)`() {
|
||||
val fact = Fact(Structure(Atom("likes"), listOf(Atom("alice"), Atom("pizza"))))
|
||||
val goal = Structure(Atom("likes"), listOf(Variable("Person"), Atom("pizza")))
|
||||
|
||||
Program.load(listOf(fact))
|
||||
|
||||
val result = Program.query(goal).toList()
|
||||
|
||||
assertEquals(1, result.size, "Expected 1 result")
|
||||
assertTrue(result[0].isSuccess, "Expected success")
|
||||
val subs = result[0].getOrNull()!!
|
||||
assertEquals(1, subs.size, "Expected 1 substitution")
|
||||
assertEquals(Atom("alice"), subs[Variable("Person")], "Expected Person to be alice")
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `likes_food(alice)`() {
|
||||
val fact = Fact(Structure(Atom("likes"), listOf(Atom("alice"), Atom("pizza"))))
|
||||
val rule = Rule(
|
||||
Structure(Atom("likes_food"), listOf(Variable("Person"))),
|
||||
Structure(Atom("likes"), listOf(Variable("Person"), Atom("pizza")))
|
||||
)
|
||||
|
||||
val goal = Structure(Atom("likes_food"), listOf(Atom("alice")))
|
||||
|
||||
Program.load(listOf(fact, rule))
|
||||
|
||||
val result = Program.query(goal).toList()
|
||||
|
||||
assertEquals(1, result.size, "Expected 1 result")
|
||||
assertTrue(result[0].isSuccess, "Expected success")
|
||||
val subs = result[0].getOrNull()!!
|
||||
assertEquals(0, subs.size, "Expected no substitutions")
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `likes_food(Person)`() {
|
||||
val fact = Fact(Structure(Atom("likes"), listOf(Atom("alice"), Atom("pizza"))))
|
||||
val rule = Rule(
|
||||
Structure(Atom("likes_food"), listOf(Variable("Person"))),
|
||||
Structure(Atom("likes"), listOf(Variable("Person"), Atom("pizza")))
|
||||
)
|
||||
|
||||
val goal = Structure(Atom("likes"), listOf(Variable("X"), Atom("pizza")))
|
||||
|
||||
Program.load(listOf(fact, rule))
|
||||
|
||||
val result = Program.query(goal).toList()
|
||||
|
||||
assertEquals(1, result.size, "Expected 1 result")
|
||||
assertTrue(result[0].isSuccess, "Expected success")
|
||||
val subs = result[0].getOrNull()!!
|
||||
assertEquals(1, subs.size, "Expected 1 substitution")
|
||||
assertEquals(Atom("alice"), subs[Variable("X")], "Expected Person to be alice")
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `requires querying exact`() {
|
||||
val fact1 = Fact(Atom("a"))
|
||||
val fact2 = Fact(Atom("b"))
|
||||
val rule1 = Rule(
|
||||
Atom("c"),
|
||||
Conjunction(
|
||||
Atom("a"),
|
||||
Atom("b")
|
||||
)
|
||||
)
|
||||
|
||||
Program.load(listOf(fact1, fact2, rule1))
|
||||
|
||||
val result = Program.query(Atom("c")).toList()
|
||||
|
||||
assertEquals(1, result.size, "Expected 1 result")
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `requires querying with variable`() {
|
||||
val fact1 = Fact(Atom("a"))
|
||||
val fact2 = Fact(Atom("b"))
|
||||
val rule1 = Rule(
|
||||
Structure(Atom("has fact"), listOf(Variable("X"))),
|
||||
Variable("X")
|
||||
)
|
||||
|
||||
Program.load(listOf(fact1, fact2, rule1))
|
||||
|
||||
val result = Program.query(Structure(Atom("has fact"), listOf(Atom("a")))).toList()
|
||||
|
||||
assertEquals(1, result.size, "Expected 1 result")
|
||||
assertTrue(result[0].isSuccess, "Expected success")
|
||||
val subs = result[0].getOrNull()!!
|
||||
assertEquals(0, subs.size, "Expected no substitutions")
|
||||
}
|
||||
|
||||
@Nested
|
||||
class `requires querying with filled variable` {
|
||||
@BeforeEach
|
||||
fun setup() {
|
||||
val fact1 = Fact(Structure(Atom("likes"), listOf(Atom("alice"), Atom("pizza"))))
|
||||
val fact2 = Fact(Structure(Atom("likes"), listOf(Atom("alice"), Atom("pasta"))))
|
||||
val fact3 = Fact(Structure(Atom("likes"), listOf(Atom("bob"), Atom("pasta"))))
|
||||
val rule1 = Rule(
|
||||
Structure(Atom("likes_italian_food"), listOf(Variable("Person"))),
|
||||
Disjunction(
|
||||
Structure(Atom("likes"), listOf(Variable("Person"), Atom("pizza"))),
|
||||
Structure(Atom("likes"), listOf(Variable("Person"), Atom("pasta")))
|
||||
)
|
||||
)
|
||||
|
||||
Program.clear()
|
||||
Program.load(listOf(fact1, fact2, fact3, rule1))
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `likes_italian_food(alice)`() {
|
||||
val result = Program.query(Structure(Atom("likes_italian_food"), listOf(Atom("alice")))).toList()
|
||||
|
||||
assertEquals(2, result.size, "Expected 2 results")
|
||||
|
||||
assertTrue(result[0].isSuccess, "Expected success")
|
||||
val subs1 = result[0].getOrNull()!!
|
||||
assertEquals(0, subs1.size, "Expected no substitutions")
|
||||
|
||||
assertTrue(result[1].isSuccess, "Expected success")
|
||||
val subs2 = result[1].getOrNull()!!
|
||||
assertEquals(0, subs2.size, "Expected no substitutions")
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `likes_italian_food(X)`() {
|
||||
val result = Program.query(Structure(Atom("likes_italian_food"), listOf(Variable("X")))).toList()
|
||||
|
||||
assertEquals(3, result.size, "Expected 3 results")
|
||||
|
||||
assertTrue(result[0].isSuccess, "Expected success")
|
||||
val subs3 = result[0].getOrNull()!!
|
||||
assertEquals(1, subs3.size, "Expected 1 substitution, especially without 'Person'")
|
||||
assertEquals(Atom("alice"), subs3[Variable("X")], "Expected alice")
|
||||
|
||||
assertTrue(result[1].isSuccess, "Expected success")
|
||||
val subs4 = result[1].getOrNull()!!
|
||||
assertEquals(1, subs4.size, "Expected 1 substitution, especially without 'Person'")
|
||||
assertEquals(Atom("alice"), subs4[Variable("X")], "Expected alice")
|
||||
|
||||
assertTrue(result[2].isSuccess, "Expected success")
|
||||
val subs5 = result[2].getOrNull()!!
|
||||
assertEquals(1, subs5.size, "Expected 1 substitution, especially without 'Person'")
|
||||
assertEquals(Atom("bob"), subs5[Variable("X")], "Expected bob")
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
Reference in a new issue