From e1ee00037f0e5ff7d90bf41f06855fd1507ac0c3 Mon Sep 17 00:00:00 2001 From: "Andreas Koepf (aider)" Date: Mon, 3 Feb 2025 22:59:23 +0100 Subject: [PATCH] test: Add deterministic test for ZebraDataset generation --- tests/test_zebra.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/tests/test_zebra.py b/tests/test_zebra.py index d233c438..054fad22 100644 --- a/tests/test_zebra.py +++ b/tests/test_zebra.py @@ -3,9 +3,18 @@ import pytest from reasoning_gym.logic.zebra_puzzles import ZebraConfig, ZebraDataset +def test_zebra_deterministic(): + """Test that dataset generates same items with same seed""" + config = ZebraConfig(seed=42, size=10, num_people=4, num_characteristics=4) + dataset1 = ZebraDataset(config) + dataset2 = ZebraDataset(config) + + for i in range(len(dataset1)): + assert dataset1[i] == dataset2[i] + + def test_zebra_puzzles(): """Test basic properties and solution of generated items""" - config = ZebraConfig(seed=42, size=10, num_people=4, num_characteristics=4) dataset = ZebraDataset(config)