Migrate tests to new API

This commit is contained in:
bakaq
2024-10-12 17:41:48 -03:00
parent 2f82c78bc0
commit e21c772181
3 changed files with 247 additions and 266 deletions

View File

@@ -1,10 +1,10 @@
use super::*; use super::*;
use crate::machine::{QueryMatch, QueryResolution, Term}; use crate::MachineBuilder;
#[test] #[test]
#[cfg_attr(miri, ignore = "it takes too long to run")] #[cfg_attr(miri, ignore = "it takes too long to run")]
fn programatic_query() { fn programatic_query() {
let mut machine = Machine::new_lib(); let mut machine = MachineBuilder::default().build();
machine.load_module_string( machine.load_module_string(
"facts", "facts",
@@ -16,39 +16,36 @@ fn programatic_query() {
), ),
); );
let query = String::from(r#"triple("a",P,"b")."#); let query = r#"triple("a",P,"b")."#;
let output = machine.run_query(query); let complete_answer: Vec<_> = machine.run_query(query).collect::<Result<_, _>>().unwrap();
assert_eq!(
output,
Ok(QueryResolution::Matches(vec![
QueryMatch::from(btreemap! {
"P" => Term::from("p1"),
}),
QueryMatch::from(btreemap! {
"P" => Term::from("p2"),
}),
]))
);
assert_eq!( assert_eq!(
machine.run_query(String::from(r#"triple("a","p1","b")."#)), complete_answer,
Ok(QueryResolution::True) [
LeafAnswer::from_bindings([("P", Term::string("p1")),]),
LeafAnswer::from_bindings([("P", Term::string("p2")),]),
],
); );
assert_eq!( let query = r#"triple("a","p1","b")."#;
machine.run_query(String::from(r#"triple("x","y","z")."#)), let complete_answer: Vec<_> = machine.run_query(query).collect::<Result<_, _>>().unwrap();
Ok(QueryResolution::False)
); assert_eq!(complete_answer, [LeafAnswer::True],);
let query = r#"triple("x","y","z")."#;
let complete_answer: Vec<_> = machine.run_query(query).collect::<Result<_, _>>().unwrap();
assert_eq!(complete_answer, [LeafAnswer::False],);
} }
#[test] #[test]
#[cfg_attr(miri, ignore = "it takes too long to run")] #[cfg_attr(miri, ignore = "it takes too long to run")]
fn failing_query() { fn failing_query() {
let mut machine = Machine::new_lib(); let mut machine = MachineBuilder::default().build();
let query = String::from(r#"triple("a",P,"b")."#); let query = r#"triple("a",P,"b")."#;
let output = machine.run_query(query); let complete_answer: Result<Vec<_>, _> = machine.run_query(query).collect();
assert_eq!( assert_eq!(
output, complete_answer,
Err(String::from( Err(String::from(
"error existence_error procedure / triple 3 / triple 3" "error existence_error procedure / triple 3 / triple 3"
)) ))
@@ -58,7 +55,7 @@ fn failing_query() {
#[test] #[test]
#[cfg_attr(miri, ignore = "it takes too long to run")] #[cfg_attr(miri, ignore = "it takes too long to run")]
fn complex_results() { fn complex_results() {
let mut machine = Machine::new_lib(); let mut machine = MachineBuilder::default().build();
machine.load_module_string( machine.load_module_string(
"facts", "facts",
r#" r#"
@@ -70,155 +67,151 @@ fn complex_results() {
subject_class("Recipe", xyz). subject_class("Recipe", xyz).
constructor(xyz, '[{action: "addLink", source: "this", predicate: "recipe://title", target: "literal://string:Meta%20Muffins"}]'). constructor(xyz, '[{action: "addLink", source: "this", predicate: "recipe://title", target: "literal://string:Meta%20Muffins"}]').
"#.to_string()); "#,
let result = machine.run_query(String::from(
"subject_class(\"Todo\", C), constructor(C, Actions).",
));
assert_eq!(
result,
Ok(QueryResolution::Matches(vec![QueryMatch::from(
btreemap! {
"C" => Term::Atom("c".into()),
"Actions" => Term::Atom("[{action: \"addLink\", source: \"this\", predicate: \"todo://state\", target: \"todo://ready\"}]".into()),
}
),]))
); );
let result = machine.run_query(String::from( let complete_answer: Vec<_> = machine
"subject_class(\"Recipe\", C), constructor(C, Actions).", .run_query(r#"subject_class("Todo", C), constructor(C, Actions)."#)
)); .collect::<Result<_, _>>()
.unwrap();
assert_eq!( assert_eq!(
result, complete_answer,
Ok(QueryResolution::Matches(vec![QueryMatch::from( [LeafAnswer::from_bindings([
btreemap! { ("C", Term::atom("c")),
"C" => Term::Atom("xyz".into()), (
"Actions" => Term::Atom("[{action: \"addLink\", source: \"this\", predicate: \"recipe://title\", target: \"literal://string:Meta%20Muffins\"}]".into()), "Actions",
} Term::atom(
),])) r#"[{action: "addLink", source: "this", predicate: "todo://state", target: "todo://ready"}]"#
)
),
])],
); );
let result = machine.run_query(String::from("subject_class(Class, _).")); let complete_answer: Vec<_> = machine
.run_query(r#"subject_class("Recipe", C), constructor(C, Actions)."#)
.collect::<Result<_, _>>()
.unwrap();
assert_eq!( assert_eq!(
result, complete_answer,
Ok(QueryResolution::Matches(vec![ [LeafAnswer::from_bindings([
QueryMatch::from(btreemap! { ("C", Term::atom("xyz")),
"Class" => Term::String("Todo".into()) (
}), "Actions",
QueryMatch::from(btreemap! { Term::atom(
"Class" => Term::String("Recipe".into()) r#"[{action: "addLink", source: "this", predicate: "recipe://title", target: "literal://string:Meta%20Muffins"}]"#
}), )
])) ),
])],
);
let complete_answer: Vec<_> = machine
.run_query("subject_class(Class, _).")
.collect::<Result<_, _>>()
.unwrap();
assert_eq!(
complete_answer,
[
LeafAnswer::from_bindings([("Class", Term::string("Todo"))]),
LeafAnswer::from_bindings([("Class", Term::string("Recipe"))]),
],
); );
} }
#[test] #[test]
#[cfg_attr(miri, ignore = "it takes too long to run")] #[cfg_attr(miri, ignore = "it takes too long to run")]
fn empty_predicate() { fn empty_predicate() {
let mut machine = Machine::new_lib(); let mut machine = MachineBuilder::default().build();
machine.load_module_string( machine.load_module_string(
"facts", "facts",
r#" r#"
:- discontiguous(subject_class/2). :- discontiguous(subject_class/2).
"# "#,
.to_string(),
); );
let result = machine.run_query(String::from("subject_class(X, _).")); let complete_answer: Vec<_> = machine
assert_eq!(result, Ok(QueryResolution::False)); .run_query("subject_class(X, _).")
.collect::<Result<_, _>>()
.unwrap();
assert_eq!(complete_answer, [LeafAnswer::False]);
} }
#[test] #[test]
#[cfg_attr(miri, ignore = "it takes too long to run")] #[cfg_attr(miri, ignore = "it takes too long to run")]
fn list_results() { fn list_results() {
let mut machine = Machine::new_lib(); let mut machine = MachineBuilder::default().build();
machine.load_module_string( machine.load_module_string(
"facts", "facts",
r#" r#"
list([1,2,3]). list([1,2,3]).
"# "#,
.to_string(),
); );
let result = machine.run_query(String::from("list(X).")); let complete_answer: Vec<_> = machine
.run_query("list(X).")
.collect::<Result<_, _>>()
.unwrap();
assert_eq!( assert_eq!(
result, complete_answer,
Ok(QueryResolution::Matches(vec![QueryMatch::from( [LeafAnswer::from_bindings([(
btreemap! { "X",
"X" => Term::List(vec![ Term::list([Term::integer(1), Term::integer(2), Term::integer(3)]),
Term::Integer(1.into()), )])],
Term::Integer(2.into()),
Term::Integer(3.into()),
]),
}
),]))
); );
} }
#[test] #[test]
#[cfg_attr(miri, ignore = "it takes too long to run")] #[cfg_attr(miri, ignore = "it takes too long to run")]
fn consult() { fn consult() {
let mut machine = Machine::new_lib(); let mut machine = MachineBuilder::default().build();
machine.consult_module_string( machine.consult_module_string(
"facts", "facts",
String::from(
r#" r#"
triple("a", "p1", "b"). triple("a", "p1", "b").
triple("a", "p2", "b"). triple("a", "p2", "b").
"#, "#,
),
); );
let query = String::from(r#"triple("a",P,"b")."#); let query = r#"triple("a",P,"b")."#;
let output = machine.run_query(query); let complete_answer: Vec<_> = machine.run_query(query).collect::<Result<_, _>>().unwrap();
assert_eq!( assert_eq!(
output, complete_answer,
Ok(QueryResolution::Matches(vec![ [
QueryMatch::from(btreemap! { LeafAnswer::from_bindings([("P", Term::string("p1"))]),
"P" => Term::from("p1"), LeafAnswer::from_bindings([("P", Term::string("p2"))]),
}), ],
QueryMatch::from(btreemap! {
"P" => Term::from("p2"),
}),
]))
); );
assert_eq!( let query = r#"triple("a","p1","b")."#;
machine.run_query(String::from(r#"triple("a","p1","b")."#)), let complete_answer: Vec<_> = machine.run_query(query).collect::<Result<_, _>>().unwrap();
Ok(QueryResolution::True) assert_eq!(complete_answer, [LeafAnswer::True],);
);
assert_eq!( let query = r#"triple("x","y","z")."#;
machine.run_query(String::from(r#"triple("x","y","z")."#)), let complete_answer: Vec<_> = machine.run_query(query).collect::<Result<_, _>>().unwrap();
Ok(QueryResolution::False) assert_eq!(complete_answer, [LeafAnswer::False],);
);
machine.consult_module_string( machine.consult_module_string(
"facts", "facts",
String::from(
r#" r#"
triple("a", "new", "b"). triple("a", "new", "b").
"#, "#,
),
); );
assert_eq!( let query = r#"triple("a","p1","b")."#;
machine.run_query(String::from(r#"triple("a","p1","b")."#)), let complete_answer: Vec<_> = machine.run_query(query).collect::<Result<_, _>>().unwrap();
Ok(QueryResolution::False) assert_eq!(complete_answer, [LeafAnswer::False],);
);
assert_eq!( let query = r#"triple("a","new","b")."#;
machine.run_query(String::from(r#"triple("a","new","b")."#)), let complete_answer: Vec<_> = machine.run_query(query).collect::<Result<_, _>>().unwrap();
Ok(QueryResolution::True) assert_eq!(complete_answer, [LeafAnswer::True]);
);
} }
/*
#[test] #[test]
#[cfg_attr(miri, ignore = "it takes too long to run")] #[cfg_attr(miri, ignore = "it takes too long to run")]
#[ignore = "uses old flawed interface"] #[ignore = "uses old flawed interface"]
fn integration_test() { fn integration_test() {
let mut machine = Machine::new_lib(); let mut machine = MachineBuilder::default().build();
// File with test commands, i.e. program code to consult and queries to run // File with test commands, i.e. program code to consult and queries to run
let code = include_str!("./lib_integration_test_commands.txt"); let code = include_str!("./lib_integration_test_commands.txt");
@@ -257,137 +250,122 @@ fn integration_test() {
} }
} }
} }
*/
#[test] #[test]
#[cfg_attr(miri, ignore = "it takes too long to run")] #[cfg_attr(miri, ignore = "it takes too long to run")]
fn findall() { fn findall() {
let mut machine = Machine::new_lib(); let mut machine = MachineBuilder::default().build();
machine.consult_module_string( machine.consult_module_string(
"facts", "facts",
String::from(
r#" r#"
triple("a", "p1", "b"). triple("a", "p1", "b").
triple("a", "p2", "b"). triple("a", "p2", "b").
"#, "#,
),
); );
let query = let query = r#"findall([Predicate, Target], triple(_,Predicate,Target), Result)."#;
String::from(r#"findall([Predicate, Target], triple(_,Predicate,Target), Result)."#); let complete_answer: Vec<_> = machine.run_query(query).collect::<Result<_, _>>().unwrap();
let output = machine.run_query(query);
assert_eq!( assert_eq!(
output, complete_answer,
Ok(QueryResolution::Matches(vec![QueryMatch::from( [LeafAnswer::from_bindings([(
btreemap! { "Result",
"Result" => Term::List( Term::list([
Vec::from([ Term::list([Term::string("p1"), Term::string("b")]),
Term::List([Term::from("p1"), Term::from("b")].into()), Term::list([Term::string("p2"), Term::string("b")]),
Term::List([Term::from("p2"), Term::from("b")].into()),
]) ])
), )])]
}
),]))
); );
} }
#[test] #[test]
#[cfg_attr(miri, ignore = "it takes too long to run")] #[cfg_attr(miri, ignore = "it takes too long to run")]
fn dont_return_partial_matches() { fn dont_return_partial_matches() {
let mut machine = Machine::new_lib(); let mut machine = MachineBuilder::default().build();
machine.consult_module_string( machine.consult_module_string(
"facts", "facts",
String::from(
r#" r#"
:- discontiguous(property_resolve/2). :- discontiguous(property_resolve/2).
subject_class("Todo", c). subject_class("Todo", c).
"#, "#,
),
); );
let query = String::from(r#"property_resolve(C, "isLiked"), subject_class("Todo", C)."#); let query = r#"property_resolve(C, "isLiked"), subject_class("Todo", C)."#;
let output = machine.run_query(query); let complete_answer: Vec<_> = machine.run_query(query).collect::<Result<_, _>>().unwrap();
assert_eq!(output, Ok(QueryResolution::False)); assert_eq!(complete_answer, [LeafAnswer::False]);
let query = String::from(r#"subject_class("Todo", C), property_resolve(C, "isLiked")."#); let query = r#"subject_class("Todo", C), property_resolve(C, "isLiked")."#;
let output = machine.run_query(query); let complete_answer: Vec<_> = machine.run_query(query).collect::<Result<_, _>>().unwrap();
assert_eq!(output, Ok(QueryResolution::False)); assert_eq!(complete_answer, [LeafAnswer::False]);
} }
#[test] #[test]
#[cfg_attr(miri, ignore = "it takes too long to run")] #[cfg_attr(miri, ignore = "it takes too long to run")]
fn dont_return_partial_matches_without_discountiguous() { fn dont_return_partial_matches_without_discountiguous() {
let mut machine = Machine::new_lib(); let mut machine = MachineBuilder::default().build();
machine.consult_module_string( machine.consult_module_string(
"facts", "facts",
String::from(
r#" r#"
a("true for a"). a("true for a").
b("true for b"). b("true for b").
"#, "#,
),
); );
let query = String::from(r#"a("true for a")."#); let query = r#"a("true for a")."#;
let output = machine.run_query(query); let complete_answer: Vec<_> = machine.run_query(query).collect::<Result<_, _>>().unwrap();
assert_eq!(output, Ok(QueryResolution::True)); assert_eq!(complete_answer, [LeafAnswer::True]);
let query = String::from(r#"a("true for a"), b("true for b")."#); let query = r#"a("true for a"), b("true for b")."#;
let output = machine.run_query(query); let complete_answer: Vec<_> = machine.run_query(query).collect::<Result<_, _>>().unwrap();
assert_eq!(output, Ok(QueryResolution::True)); assert_eq!(complete_answer, [LeafAnswer::True]);
let query = String::from(r#"a("true for b"), b("true for b")."#); let query = r#"a("true for b"), b("true for b")."#;
let output = machine.run_query(query); let complete_answer: Vec<_> = machine.run_query(query).collect::<Result<_, _>>().unwrap();
assert_eq!(output, Ok(QueryResolution::False)); assert_eq!(complete_answer, [LeafAnswer::False]);
let query = String::from(r#"a("true for a"), b("true for a")."#); let query = r#"a("true for a"), b("true for a")."#;
let output = machine.run_query(query); let complete_answer: Vec<_> = machine.run_query(query).collect::<Result<_, _>>().unwrap();
assert_eq!(output, Ok(QueryResolution::False)); assert_eq!(complete_answer, [LeafAnswer::False]);
} }
#[test] #[test]
#[cfg_attr(miri, ignore = "it takes too long to run")] #[cfg_attr(miri, ignore = "it takes too long to run")]
fn non_existent_predicate_should_not_cause_panic_when_other_predicates_are_defined() { fn non_existent_predicate_should_not_cause_panic_when_other_predicates_are_defined() {
let mut machine = Machine::new_lib(); let mut machine = MachineBuilder::default().build();
machine.consult_module_string( machine.consult_module_string(
"facts", "facts",
String::from(
r#" r#"
triple("a", "p1", "b"). triple("a", "p1", "b").
triple("a", "p2", "b"). triple("a", "p2", "b").
"#, "#,
),
); );
let query = String::from("non_existent_predicate(\"a\",\"p1\",\"b\")."); let query = r#"non_existent_predicate("a","p1","b")."#;
let complete_answer: Result<Vec<_>, _> = machine.run_query(query).collect();
let result = machine.run_query(query);
assert_eq!( assert_eq!(
result, complete_answer,
Err(String::from("error existence_error procedure / non_existent_predicate 3 / non_existent_predicate 3")) Err(String::from(
"error existence_error procedure / non_existent_predicate 3 / non_existent_predicate 3"
))
); );
} }
#[test] #[test]
#[cfg_attr(miri, ignore)] #[cfg_attr(miri, ignore)]
fn atom_quoting() { fn atom_quoting() {
let mut machine = Machine::new_lib(); let mut machine = MachineBuilder::default().build();
let query = "X = '.'.".into(); let query = "X = '.'.";
let complete_answer: Vec<_> = machine.run_query(query).collect::<Result<_, _>>().unwrap();
let result = machine.run_query(query);
assert_eq!( assert_eq!(
result, complete_answer,
Ok(QueryResolution::Matches(vec![QueryMatch::from( [LeafAnswer::from_bindings([("X", Term::atom("."))])]
btreemap! {
"X" => Term::Atom(".".into()),
}
)]))
); );
} }
@@ -395,19 +373,17 @@ fn atom_quoting() {
#[cfg_attr(miri, ignore)] #[cfg_attr(miri, ignore)]
fn rational_number() { fn rational_number() {
use crate::parser::dashu::rational::RBig; use crate::parser::dashu::rational::RBig;
let mut machine = Machine::new_lib(); let mut machine = MachineBuilder::default().build();
let query = "X is 1 rdiv 2.".into(); let query = "X is 1 rdiv 2.";
let complete_answer: Vec<_> = machine.run_query(query).collect::<Result<_, _>>().unwrap();
let result = machine.run_query(query);
assert_eq!( assert_eq!(
result, complete_answer,
Ok(QueryResolution::Matches(vec![QueryMatch::from( [LeafAnswer::from_bindings([(
btreemap! { "X",
"X" => Term::Rational(RBig::from_parts(1.into(), 2u32.into())), Term::rational(RBig::from_parts(1.into(), 2u32.into()))
} )])]
)]))
); );
} }
@@ -415,33 +391,30 @@ fn rational_number() {
#[cfg_attr(miri, ignore)] #[cfg_attr(miri, ignore)]
fn big_integer() { fn big_integer() {
use crate::parser::dashu::integer::IBig; use crate::parser::dashu::integer::IBig;
let mut machine = Machine::new_lib(); let mut machine = MachineBuilder::default().build();
let query = "X is 10^100.".into(); let query = "X is 10^100.";
let complete_answer: Vec<_> = machine.run_query(query).collect::<Result<_, _>>().unwrap();
let result = machine.run_query(query);
assert_eq!( assert_eq!(
result, complete_answer,
Ok(QueryResolution::Matches(vec![QueryMatch::from( [LeafAnswer::from_bindings([(
btreemap! { "X",
"X" => Term::Integer(IBig::from(10).pow(100)), Term::integer(IBig::from(10).pow(100))
} )])],
)]))
); );
} }
#[test] #[test]
#[cfg_attr(miri, ignore)] #[cfg_attr(miri, ignore)]
fn complicated_term() { fn complicated_term() {
let mut machine = Machine::new_lib(); let mut machine = MachineBuilder::default().build();
let query = "X = a(\"asdf\", [42, 2.54, asdf, a, [a,b|_], Z]).".into(); let query = r#"X = a("asdf", [42, 2.54, asdf, a, [a,b|_], Z])."#;
let complete_answer: Vec<_> = machine.run_query(query).collect::<Result<_, _>>().unwrap();
let result = machine.run_query(query); let expected = Term::Compound(
// Compound term
let expected = Term::Structure(
// Composite term
"a".into(), "a".into(),
vec![ vec![
Term::String("asdf".into()), // String Term::String("asdf".into()), // String
@@ -450,12 +423,12 @@ fn complicated_term() {
Term::Float(2.54.into()), // Float Term::Float(2.54.into()), // Float
Term::Atom("asdf".into()), // Atom Term::Atom("asdf".into()), // Atom
Term::Atom("a".into()), // Char Term::Atom("a".into()), // Char
Term::Structure( Term::Compound(
// Partial string // Partial string
".".into(), ".".into(),
vec![ vec![
Term::Atom("a".into()), Term::Atom("a".into()),
Term::Structure( Term::Compound(
".".into(), ".".into(),
vec![ vec![
Term::Atom("b".into()), Term::Atom("b".into()),
@@ -470,51 +443,45 @@ fn complicated_term() {
); );
assert_eq!( assert_eq!(
result, complete_answer,
Ok(QueryResolution::Matches(vec![QueryMatch::from( [LeafAnswer::from_bindings([("X", expected)])]
btreemap! {
"X" => expected,
}
)]))
); );
} }
#[test] #[test]
#[cfg_attr(miri, ignore = "it takes too long to run")] #[cfg_attr(miri, ignore = "it takes too long to run")]
fn issue_2341() { fn issue_2341() {
let mut machine = Machine::new_lib(); let mut machine = MachineBuilder::default().build();
machine.load_module_string( machine.load_module_string(
"facts", "facts",
String::from(
r#" r#"
male(stephen). male(stephen).
parent(albert,edward). parent(albert,edward).
father(F,C):-parent(F,C),male(F). father(F,C):-parent(F,C),male(F).
"#, "#,
),
); );
let query = String::from(r#"father(F,C)."#); let query = r#"father(F,C)."#;
let output = machine.run_query(query); let complete_answer: Vec<_> = machine.run_query(query).collect::<Result<_, _>>().unwrap();
assert_eq!(output, Ok(QueryResolution::False)); assert_eq!(complete_answer, [LeafAnswer::False]);
} }
#[test] #[test]
#[cfg_attr(miri, ignore)] #[cfg_attr(miri, ignore)]
fn query_iterator_determinism() { fn query_iterator_determinism() {
let mut machine = Machine::new_lib(); let mut machine = MachineBuilder::default().build();
{ {
let mut iterator = machine.run_query_iter("X = 1.".into()); let mut iterator = machine.run_query("X = 1.");
iterator.next(); iterator.next();
assert_eq!(iterator.next(), None); assert_eq!(iterator.next(), None);
} }
{ {
let mut iterator = machine.run_query_iter("X = 1 ; false.".into()); let mut iterator = machine.run_query("X = 1 ; false.");
iterator.next(); iterator.next();
@@ -523,7 +490,7 @@ fn query_iterator_determinism() {
} }
{ {
let mut iterator = machine.run_query_iter("false.".into()); let mut iterator = machine.run_query("false.");
assert_eq!(iterator.next(), Some(Ok(LeafAnswer::False))); assert_eq!(iterator.next(), Some(Ok(LeafAnswer::False)));
assert_eq!(iterator.next(), None); assert_eq!(iterator.next(), None);
@@ -533,9 +500,9 @@ fn query_iterator_determinism() {
#[test] #[test]
#[cfg_attr(miri, ignore)] #[cfg_attr(miri, ignore)]
fn query_iterator_backtracking_when_no_variables() { fn query_iterator_backtracking_when_no_variables() {
let mut machine = Machine::new_lib(); let mut machine = MachineBuilder::default().build();
let mut iterator = machine.run_query_iter("true;false.".into()); let mut iterator = machine.run_query("true;false.");
assert_eq!(iterator.next(), Some(Ok(LeafAnswer::True))); assert_eq!(iterator.next(), Some(Ok(LeafAnswer::True)));
assert_eq!(iterator.next(), Some(Ok(LeafAnswer::False))); assert_eq!(iterator.next(), Some(Ok(LeafAnswer::False)));
@@ -545,39 +512,43 @@ fn query_iterator_backtracking_when_no_variables() {
#[test] #[test]
#[cfg_attr(miri, ignore)] #[cfg_attr(miri, ignore)]
fn differentiate_anonymous_variables() { fn differentiate_anonymous_variables() {
let mut machine = Machine::new_lib(); let mut machine = MachineBuilder::default().build();
let result = machine.run_query("A = [_,_], _B = 1 ; B = [_,_].".into()); let complete_answer: Vec<_> = machine
.run_query("A = [_,_], _B = 1 ; B = [_,_].")
.collect::<Result<_, _>>()
.unwrap();
assert_eq!( assert_eq!(
result, complete_answer,
Ok(QueryResolution::Matches(vec![ [
QueryMatch::from(btreemap! { LeafAnswer::from_bindings([
"A" => Term::List(vec![Term::Var("_A".into()), Term::Var("_C".into())]), (
"_B" => Term::Integer(1.into()), "A",
}), Term::list([Term::variable("_A"), Term::variable("_C")])
QueryMatch::from(btreemap! { ),
"B" => Term::List(vec![Term::Var("_A".into()), Term::Var("_C".into())]), ("_B", Term::integer(1)),
}), ]),
])) LeafAnswer::from_bindings([(
"B",
Term::list([Term::variable("_A"), Term::variable("_C")])
),]),
]
); );
} }
#[test] #[test]
#[cfg_attr(miri, ignore)] #[cfg_attr(miri, ignore)]
fn order_of_variables_in_binding() { fn order_of_variables_in_binding() {
let mut machine = Machine::new_lib(); let mut machine = MachineBuilder::default().build();
let result = machine.run_query("X = Y, Z = W.".into()); let complete_answer: Vec<_> = machine.run_query("X = Y, Z = W.").collect::<Result<_,_>>().unwrap();
assert_eq!( assert_eq!(
result, complete_answer,
Ok(QueryResolution::Matches(vec![QueryMatch::from( [LeafAnswer::from_bindings([
btreemap! { ("X", Term::variable("Y")),
"X" => Term::Var("Y".into()), ("Z", Term::variable("W")),
"Z" => Term::Var("W".into()), ])]
}
),]))
); );
} }

View File

@@ -44,6 +44,16 @@ pub enum LeafAnswer {
}, },
} }
impl LeafAnswer {
/// Creates a leaf answer with no residual goals.
pub fn from_bindings<S: Into<String>>(bindings: impl IntoIterator<Item = (S, Term)>) -> Self {
LeafAnswer::LeafAnswer {
bindings: bindings.into_iter().map(|(k, v)| (k.into(), v)).collect(),
residual_goals: Vec::new(),
}
}
}
/// Represents a Prolog term. /// Represents a Prolog term.
#[non_exhaustive] #[non_exhaustive]
#[derive(Debug, Clone, PartialEq)] #[derive(Debug, Clone, PartialEq)]

View File

@@ -26,8 +26,8 @@ impl Expectable for &[u8] {
/// Tests whether the file can be successfully loaded /// Tests whether the file can be successfully loaded
/// and produces the expected output during it /// and produces the expected output during it
pub(crate) fn load_module_test<T: Expectable>(file: &str, expected: T) { pub(crate) fn load_module_test<T: Expectable>(file: &str, expected: T) {
use scryer_prolog::Machine; use scryer_prolog::MachineBuilder;
let mut wam = Machine::with_test_streams(); let mut wam = MachineBuilder::default().build();
expected.assert_eq(wam.test_load_file(file).as_slice()); expected.assert_eq(wam.test_load_file(file).as_slice());
} }