| @@ -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]] | ||||