Browse Source

[ENH] update test for reasoning

pull/1/head
Tony-HYX 1 year ago
parent
commit
fbb3ff2a3a
1 changed files with 7 additions and 1 deletions
  1. +7
    -1
      tests/test_reasoning.py

+ 7
- 1
tests/test_reasoning.py View File

@@ -16,10 +16,16 @@ class TestKBBase(object):
assert result == 3 assert result == 3


def test_revise_at_idx(self, kb_add): 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]] 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): 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) result = kb_add.abduce_candidates([1, 2], 1, max_revision_num=2, require_more_revision=0)
assert result == [[1, 0]] assert result == [[1, 0]]




Loading…
Cancel
Save