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

Refactor CaesarCipher

Browse source
This commit is contained in:
EduardDurech 2025-02-09 05:15:12 +00:00
parent 23aa6ca7e7
commit 5279ccf7e1
8 changed files with 513 additions and 159 deletions
Download patch file Download diff file Expand all files Collapse all files

97
examples/exercises/algorithmic/caesar_cipher_examples.py Normal file
View file

@ -0,0 +1,97 @@
"""Examples of generated problems from the CaesarCipher exercise.
This file demonstrates different types of Caesar cipher problems that can be generated
at various difficulty levels.
"""
from reasoning_gym.curricula.algorithmic.caesar_cipher_curriculum import CaesarCipherCurriculum
from reasoning_gym.exercises.algorithmic.caesar_cipher import CaesarCipherExercise
import random
def main():
# Initialize with fixed seed for reproducibility
curriculum = CaesarCipherCurriculum()
exercise = CaesarCipherExercise()
curriculum.rng = random.Random(42)
print("\n========================================\n")
# Level 0: Basic decryption with short text and small rotation
curriculum.set_attr_level("num_words", 0) # Short text (5 words)
curriculum.set_attr_level("rotation", 0) # Small rotation (1-3)
curriculum.set_attr_level("text_case", 0) # UPPER case only
problem = exercise.generate(curriculum)
print("Level 0 (Basic Decryption):")
print(problem)
print("\n========================================\n")
# Level 1: Medium length text with larger rotation
curriculum.set_attr_level("num_words", 1) # Medium text (10 words)
curriculum.set_attr_level("rotation", 2) # Medium rotation (10-15)
curriculum.set_attr_level("text_case", 1) # lower case only
problem = exercise.generate(curriculum)
print("Level 1 (Medium Length Text):")
print(problem)
print("\n========================================\n")
# Level 2: Long text with mixed case and large rotation
curriculum.set_attr_level("num_words", 2) # Long text (20 words)
curriculum.set_attr_level("rotation", 4) # Large rotation (20-25)
curriculum.set_attr_level("text_case", 2) # Mixed case with preserved capitalization
problem = exercise.generate(curriculum)
print("Level 2 (Complex Text):")
print(problem)
print("\n========================================\n")
# Random Examples with Different Seeds
print("Random Examples (Different Seeds):")
for seed in range(10, 15):
curriculum.rng = random.Random(seed)
# Randomly set curriculum levels
curriculum.set_attr_level("num_words", random.randint(0, 2))
curriculum.set_attr_level("rotation", random.randint(0, 4))
curriculum.set_attr_level("text_case", random.randint(0, 2))
problem = exercise.generate(curriculum)
print(f"\nRandom Example (Seed {seed}):")
print(problem)
print("\n========================================\n")
# Special Cases
print("Special Cases:")
# Case 1: Maximum length with small rotation
for attr in curriculum.attributes: # Reset all attributes
curriculum.set_attr_level(attr, 0)
curriculum.set_attr_level("num_words", 2) # Maximum words (20)
curriculum.set_attr_level("rotation", 0) # Small rotation (1-3)
curriculum.set_attr_level("text_case", 0) # UPPER case
problem = exercise.generate(curriculum)
print("\nLong Text with Small Rotation:")
print(problem)
# Case 2: Short text with maximum rotation
for attr in curriculum.attributes: # Reset all attributes
curriculum.set_attr_level(attr, 0)
curriculum.set_attr_level("num_words", 0) # Minimum words (5)
curriculum.set_attr_level("rotation", 4) # Maximum rotation (20-25)
curriculum.set_attr_level("text_case", 2) # Mixed case
problem = exercise.generate(curriculum)
print("\nShort Text with Large Rotation:")
print(problem)
# Case 3: Medium text with mixed case
for attr in curriculum.attributes: # Reset all attributes
curriculum.set_attr_level(attr, 0)
curriculum.set_attr_level("num_words", 1) # Medium length (10 words)
curriculum.set_attr_level("rotation", 2) # Medium rotation (10-15)
curriculum.set_attr_level("text_case", 2) # Mixed case
problem = exercise.generate(curriculum)
print("\nMedium Text with Mixed Case:")
print(problem)
if __name__ == "__main__":
main()