Add a few property-based tests

This commit is contained in:
Hassan Abedi 2026-03-09 17:42:33 +01:00
parent dda7bfb9e8
commit a8ac280b02
3 changed files with 394 additions and 0 deletions

1
.gitignore vendored
View File

@ -81,3 +81,4 @@ Cargo.lock
.benchmarks
.env
.claude/
*.proptest-regressions

View File

@ -32,6 +32,9 @@ binaries = []
[dependencies]
[dev-dependencies]
proptest = "1.6"
[profile.release]
strip = "debuginfo"
panic = "unwind"

View File

@ -1 +1,391 @@
//! Property-based tests for the chase algorithm using proptest.
use proptest::prelude::*;
use chase_rs::chase::rule::RuleBuilder;
use chase_rs::{Atom, Instance, Term, chase};
/// Strategy for generating constant names
fn constant_strategy() -> impl Strategy<Value = String> {
prop::sample::select(vec![
"a".to_string(),
"b".to_string(),
"c".to_string(),
"d".to_string(),
"e".to_string(),
])
}
/// Strategy for generating variable names
fn variable_strategy() -> impl Strategy<Value = String> {
prop::sample::select(vec![
"X".to_string(),
"Y".to_string(),
"Z".to_string(),
"W".to_string(),
])
}
/// Strategy for generating predicate names
fn predicate_strategy() -> impl Strategy<Value = String> {
prop::sample::select(vec![
"P".to_string(),
"Q".to_string(),
"R".to_string(),
"S".to_string(),
])
}
/// Strategy for generating ground atoms (facts)
fn ground_atom_strategy() -> impl Strategy<Value = Atom> {
(
predicate_strategy(),
prop::collection::vec(constant_strategy(), 1..=3),
)
.prop_map(|(pred, consts)| {
Atom::new(pred, consts.into_iter().map(Term::constant).collect())
})
}
/// Strategy for generating a small instance (set of facts)
fn instance_strategy() -> impl Strategy<Value = Instance> {
prop::collection::vec(ground_atom_strategy(), 0..10)
.prop_map(|atoms| atoms.into_iter().collect())
}
// =============================================================================
// Property: Chase always terminates
// =============================================================================
proptest! {
#![proptest_config(ProptestConfig::with_cases(50))]
#[test]
fn chase_always_terminates(instance in instance_strategy()) {
// Simple rule: P(X) -> Q(X)
let rule = RuleBuilder::new()
.when("P", vec![Term::var("X")])
.then("Q", vec![Term::var("X")])
.build();
let result = chase(instance, &[rule]);
// Chase should always terminate (either fixpoint or max iterations)
prop_assert!(result.terminated || result.steps > 0);
}
}
// =============================================================================
// Property: Chase is monotonic (facts are only added, never removed)
// =============================================================================
proptest! {
#![proptest_config(ProptestConfig::with_cases(50))]
#[test]
fn chase_is_monotonic(instance in instance_strategy()) {
let initial_facts: Vec<_> = instance.iter().cloned().collect();
let rule = RuleBuilder::new()
.when("P", vec![Term::var("X")])
.then("Q", vec![Term::var("X")])
.build();
let result = chase(instance, &[rule]);
// All original facts should still be present
for fact in &initial_facts {
prop_assert!(
result.instance.contains(fact),
"Original fact {:?} was removed", fact
);
}
// Result should have at least as many facts as the input
prop_assert!(result.instance.len() >= initial_facts.len());
}
}
// =============================================================================
// Property: Fixpoint is stable (running chase again produces no changes)
// =============================================================================
proptest! {
#![proptest_config(ProptestConfig::with_cases(30))]
#[test]
fn fixpoint_is_stable(instance in instance_strategy()) {
let rule = RuleBuilder::new()
.when("P", vec![Term::var("X")])
.then("Q", vec![Term::var("X")])
.build();
// Run chase to fixpoint
let result1 = chase(instance, &[rule.clone()]);
prop_assert!(result1.terminated);
// Run chase again on the result
let result2 = chase(result1.instance.clone(), &[rule]);
// Should terminate immediately with no new facts
prop_assert!(result2.terminated);
prop_assert_eq!(result2.steps, 0, "Second chase should produce no changes");
prop_assert_eq!(result1.instance.len(), result2.instance.len());
}
}
// =============================================================================
// Property: All derived facts are ground (no variables)
// =============================================================================
proptest! {
#![proptest_config(ProptestConfig::with_cases(50))]
#[test]
fn all_facts_are_ground(instance in instance_strategy()) {
let rule = RuleBuilder::new()
.when("P", vec![Term::var("X"), Term::var("Y")])
.then("R", vec![Term::var("X"), Term::var("Y")])
.build();
let result = chase(instance, &[rule]);
for fact in result.instance.iter() {
prop_assert!(
fact.is_ground(),
"Fact {:?} contains variables", fact
);
}
}
}
// =============================================================================
// Property: Rules without existentials don't create nulls
// =============================================================================
proptest! {
#![proptest_config(ProptestConfig::with_cases(50))]
#[test]
fn no_existentials_means_no_nulls(instance in instance_strategy()) {
// Rule with all head variables appearing in body
let rule = RuleBuilder::new()
.when("P", vec![Term::var("X")])
.then("Q", vec![Term::var("X")])
.build();
let result = chase(instance, &[rule]);
for fact in result.instance.iter() {
for term in &fact.terms {
prop_assert!(
!matches!(term, Term::Null(_)),
"Found null in fact {:?} but rule has no existentials", fact
);
}
}
}
}
// =============================================================================
// Property: Existential rules create exactly one null per trigger
// =============================================================================
proptest! {
#![proptest_config(ProptestConfig::with_cases(30))]
#[test]
fn existentials_create_unique_nulls(
num_constants in 1usize..5
) {
// Create facts P(c0), P(c1), P(c2), ... with unique constants
let constants: Vec<String> = (0..num_constants).map(|i| format!("c{}", i)).collect();
let instance: Instance = constants
.iter()
.map(|c| Atom::new("P", vec![Term::constant(c)]))
.collect();
// Rule: P(X) -> Q(X, Y) where Y is existential
let rule = RuleBuilder::new()
.when("P", vec![Term::var("X")])
.then("Q", vec![Term::var("X"), Term::var("Y")])
.build();
let result = chase(instance, &[rule]);
let q_facts: Vec<_> = result.instance.facts_for_predicate("Q");
// Should have exactly as many Q facts as P facts
prop_assert_eq!(q_facts.len(), num_constants);
// Each Q fact should have a null in the second position
for fact in &q_facts {
prop_assert!(
matches!(fact.terms[1], Term::Null(_)),
"Expected null in second position of {:?}", fact
);
}
// All nulls should be unique
let nulls: Vec<_> = q_facts
.iter()
.filter_map(|f| match &f.terms[1] {
Term::Null(id) => Some(*id),
_ => None,
})
.collect();
let mut unique_nulls = nulls.clone();
unique_nulls.sort();
unique_nulls.dedup();
prop_assert_eq!(nulls.len(), unique_nulls.len(), "Nulls should be unique");
}
}
// =============================================================================
// Property: Transitive closure computes correct number of facts
// =============================================================================
proptest! {
#![proptest_config(ProptestConfig::with_cases(20))]
#[test]
fn transitive_closure_chain(chain_length in 2usize..6) {
// Create a chain: E(0,1), E(1,2), ..., E(n-1,n)
let mut facts = Vec::new();
for i in 0..chain_length {
facts.push(Atom::new(
"E",
vec![Term::constant(format!("{}", i)), Term::constant(format!("{}", i + 1))],
));
}
let instance: Instance = facts.into_iter().collect();
// E(X,Y) -> T(X,Y)
let rule1 = RuleBuilder::new()
.when("E", vec![Term::var("X"), Term::var("Y")])
.then("T", vec![Term::var("X"), Term::var("Y")])
.build();
// T(X,Y), E(Y,Z) -> T(X,Z)
let rule2 = RuleBuilder::new()
.when("T", vec![Term::var("X"), Term::var("Y")])
.when("E", vec![Term::var("Y"), Term::var("Z")])
.then("T", vec![Term::var("X"), Term::var("Z")])
.build();
let result = chase(instance, &[rule1, rule2]);
prop_assert!(result.terminated);
let t_facts = result.instance.facts_for_predicate("T");
// For a chain of n edges, transitive closure has n*(n+1)/2 pairs
let expected = chain_length * (chain_length + 1) / 2;
prop_assert_eq!(
t_facts.len(),
expected,
"Chain of {} edges should have {} transitive pairs",
chain_length,
expected
);
}
}
// =============================================================================
// Property: Empty instance with any rules produces empty or stays empty
// =============================================================================
proptest! {
#[test]
fn empty_instance_stays_bounded(_seed in 0u64..1000) {
let instance = Instance::new();
let rule = RuleBuilder::new()
.when("P", vec![Term::var("X")])
.then("Q", vec![Term::var("X")])
.build();
let result = chase(instance, &[rule]);
prop_assert!(result.terminated);
prop_assert_eq!(result.steps, 0);
prop_assert!(result.instance.is_empty());
}
}
// =============================================================================
// Property: Substitution application is idempotent for ground terms
// =============================================================================
use chase_rs::Substitution;
proptest! {
#![proptest_config(ProptestConfig::with_cases(100))]
#[test]
fn substitution_ground_term_unchanged(c in constant_strategy()) {
let term = Term::constant(&c);
let mut subst = Substitution::new();
subst.bind("X".to_string(), Term::constant("other"));
let result = subst.apply_term(&term);
prop_assert_eq!(result, term, "Ground term should be unchanged by substitution");
}
}
// =============================================================================
// Property: Substitution correctly replaces variables
// =============================================================================
proptest! {
#![proptest_config(ProptestConfig::with_cases(100))]
#[test]
fn substitution_replaces_variable(
var in variable_strategy(),
replacement in constant_strategy()
) {
let term = Term::var(&var);
let mut subst = Substitution::new();
subst.bind(var.clone(), Term::constant(&replacement));
let result = subst.apply_term(&term);
prop_assert_eq!(
result,
Term::constant(&replacement),
"Variable should be replaced"
);
}
}
// =============================================================================
// Property: Atom arity is preserved through substitution
// =============================================================================
proptest! {
#![proptest_config(ProptestConfig::with_cases(50))]
#[test]
fn substitution_preserves_arity(
pred in predicate_strategy(),
vars in prop::collection::vec(variable_strategy(), 1..4),
replacement in constant_strategy()
) {
let atom = Atom::new(
&pred,
vars.iter().map(|v| Term::var(v)).collect(),
);
let mut subst = Substitution::new();
for var in &vars {
subst.bind(var.clone(), Term::constant(&replacement));
}
let result = subst.apply_atom(&atom);
prop_assert_eq!(result.arity(), atom.arity());
prop_assert_eq!(result.predicate, atom.predicate);
}
}