mirror of
https://github.com/open-thought/reasoning-gym.git
synced 2026-04-28 17:29:39 +00:00
test: Add test for ABConfig dataset generation
This commit is contained in:
parent
8eee8b3770
commit
4b7abd0ffd
1 changed files with 1 additions and 3 deletions
|
|
@ -1,9 +1,7 @@
|
||||||
import pytest
|
|
||||||
|
|
||||||
from reasoning_gym.algorithmic.ab import ABConfig, ABDataset
|
from reasoning_gym.algorithmic.ab import ABConfig, ABDataset
|
||||||
|
|
||||||
|
|
||||||
def test_zebra_puzzles():
|
def test_abconfig_puzzles():
|
||||||
"""Test basic properties and solution of generated items"""
|
"""Test basic properties and solution of generated items"""
|
||||||
config = ABConfig(seed=42, size=10, length=5)
|
config = ABConfig(seed=42, size=10, length=5)
|
||||||
dataset = ABDataset(config)
|
dataset = ABDataset(config)
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue