open-thought/reasoning-gym
1
0
Fork 0
mirror of https://github.com/open-thought/reasoning-gym.git synced 2026-04-23 16:55:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

added testing func for prop logic

Browse source
This commit is contained in:
joesharratt1229 2025-02-20 23:59:07 +00:00
parent 1996ffa6d8
commit a67851d3ea
1 changed files with 14 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

14
tests/test_propositional_logic.py
View file

@ -87,3 +87,17 @@ def test_propositional_logic_dataset_iteration():
# Test multiple iterations yield same items
assert items == list(dataset)
def test_propositional_logic_dataset_score_answer_correct():
dataset = PropositionalLogicDataset(PropositionalLogicConfig(size=50, seed=101))
for i, item in enumerate(dataset):
score = dataset.score_answer(item["metadata"]["example_answer"], item)
assert score == 1.0
def test_propositional_logic_dataset_score_answer_incorrect():
dataset = PropositionalLogicDataset(PropositionalLogicConfig(size=100, seed=101))
for i, item in enumerate(dataset):
score = dataset.score_answer("Wrong", item)
assert score == 0.01