formatting

tests/test_prime_factorization.py

@@ -1,10 +1,8 @@
 """Tests for prime factorization task generation"""
+
 import pytest
 
-from reasoning_gym.arithmetic.prime_factorization import (
-    PrimeFactorizationConfig,
-    PrimeFactorizationDataset,
-)
+from reasoning_gym.arithmetic.prime_factorization import PrimeFactorizationConfig, PrimeFactorizationDataset
 
 
 def test_prime_factorization_config_validation():
@@ -30,12 +28,7 @@ def test_prime_factorization_dataset_deterministic():
 
 def test_prime_factorization_dataset_items():
     """Test basic properties of generated items"""
-    config = PrimeFactorizationConfig(
-        min_value=2,
-        max_value=100,
-        size=10,
-        seed=42
-    )
+    config = PrimeFactorizationConfig(min_value=2, max_value=100, size=10, seed=42)
     dataset = PrimeFactorizationDataset(config)
 
     for i in range(len(dataset)):
@@ -45,26 +38,26 @@ def test_prime_factorization_dataset_items():
         assert "question" in item
         assert "answer" in item
         assert "metadata" in item
-        
+
         # Check metadata
         assert "number" in item["metadata"]
         assert "factors" in item["metadata"]
-        
+
         # Verify value range
         number = item["metadata"]["number"]
         assert config.min_value <= number <= config.max_value
-        
+
         # Verify factorization is correct
         factors = item["metadata"]["factors"]
         product = 1
         for factor in factors:
             product *= factor
         assert product == number
-        
+
         # Verify factors are prime
         for factor in factors:
             assert is_prime(factor), f"{factor} is not prime"
-        
+
         # Verify answer format
         assert item["answer"] == " × ".join(map(str, factors))
 
@@ -83,15 +76,10 @@ def test_prime_factorization_dataset_iteration():
 
 def test_prime_factorization_known_values():
     """Test factorization of known values"""
-    config = PrimeFactorizationConfig(
-        min_value=12,
-        max_value=12,  # Force specific number
-        size=1,
-        seed=42
-    )
+    config = PrimeFactorizationConfig(min_value=12, max_value=12, size=1, seed=42)  # Force specific number
     dataset = PrimeFactorizationDataset(config)
     item = dataset[0]
-    
+
     assert item["metadata"]["number"] == 12
     assert item["metadata"]["factors"] == [2, 2, 3]
     assert item["answer"] == "2 × 2 × 3"
@@ -101,7 +89,7 @@ def is_prime(n: int) -> bool:
     """Helper function to check if a number is prime"""
     if n < 2:
         return False
-    for i in range(2, int(n ** 0.5) + 1):
+    for i in range(2, int(n**0.5) + 1):
         if n % i == 0:
             return False
     return True