From fbb3ff2a3a1a06962eb6619cf8053030ac539cd7 Mon Sep 17 00:00:00 2001 From: Tony-HYX <605698554@qq.com> Date: Sat, 9 Dec 2023 01:02:21 +0800 Subject: [PATCH] [ENH] update test for reasoning --- tests/test_reasoning.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/tests/test_reasoning.py b/tests/test_reasoning.py index 95c98a2..c4d29f5 100644 --- a/tests/test_reasoning.py +++ b/tests/test_reasoning.py @@ -16,10 +16,16 @@ class TestKBBase(object): assert result == 3 def test_revise_at_idx(self, kb_add): - result = kb_add.revise_at_idx([1, 2], 2, [0]) + result = kb_add.revise_at_idx([0, 2], 2, []) assert result == [[0, 2]] + result = kb_add.revise_at_idx([1, 2], 2, []) + assert result == [] + result = kb_add.revise_at_idx([1, 2], 2, [0, 1]) + assert result == [[0, 2], [1, 1], [2, 0]] def test_abduce_candidates(self, kb_add): + result = kb_add.abduce_candidates([0, 1], 1, max_revision_num=2, require_more_revision=0) + assert result == [[0, 1]] result = kb_add.abduce_candidates([1, 2], 1, max_revision_num=2, require_more_revision=0) assert result == [[1, 0]]