exact tests

Author: KRRT7

tests/test_hypothesis_testing.py

@@ -12,8 +12,9 @@ def test_adds_derandomize_decorator():
 def test_x(x):
     assert isinstance(x, int)
 """
+    expected = """from hypothesis import settings\nfrom hypothesis import given, strategies as st\n\n@given(x=st.integers(min_value=-10000, max_value=10000))\n@settings(derandomize=True)\ndef test_x(x):\n    assert isinstance(x, int)"""
     out = make_hypothesis_tests_deterministic(src)
-    assert "@settings(derandomize=True)" in out or "settings(derandomize=True)" in out
+    assert out == expected
 
 
 def test_integers_constrained_with_negatives():
@@ -23,11 +24,9 @@ def test_integers_constrained_with_negatives():
 def t(x):
     pass
 """
+    expected = """from hypothesis import settings\nfrom hypothesis import given, strategies as st\n\n@given(x=st.integers(min_value=-10000, max_value=10000))\n@settings(derandomize=True)\ndef t(x):\n    pass"""
     out = make_hypothesis_tests_deterministic(src)
-    # Remove spaces for easier checking
-    normalized = out.replace(" ", "").replace("\n", "")
-    assert "min_value=-10000" in normalized
-    assert "max_value=10000" in normalized
+    assert out == expected
 
 
 def test_floats_constrained_to_finite():
@@ -37,11 +36,9 @@ def test_floats_constrained_to_finite():
 def t(x):
     pass
 """
+    expected = """from hypothesis import settings\nfrom hypothesis import given, strategies as st\n\n@given(x=st.floats(min_value=-1000000.0, max_value=1000000.0, allow_nan=False, allow_infinity=False))\n@settings(derandomize=True)\ndef t(x):\n    pass"""
     out = make_hypothesis_tests_deterministic(src)
-    normalized = out.replace(" ", "").replace("\n", "")
-    assert "allow_nan=False" in normalized
-    assert "allow_infinity=False" in normalized
-    assert "min_value=" in normalized and "max_value=" in normalized
+    assert out == expected
 
 
 def test_existing_constraints_not_overridden():
@@ -53,14 +50,9 @@ def test_existing_constraints_not_overridden():
 def t(x):
     pass
 """
+    expected = """from hypothesis import given, strategies as st, settings\n\n@settings(derandomize=True, max_examples=5)\n@given(x=st.integers(min_value=-5, max_value=5))\ndef t(x):\n    pass"""
     out = make_hypothesis_tests_deterministic(src)
-    # Should not add duplicate settings decorator
-    assert out.count("@settings") == 1
-    # Should preserve original constraints
-    assert "min_value=-5" in out or "min_value= -5" in out
-    assert "max_value=5" in out or "max_value= 5" in out
-    # Should not add the default -10000/10000 bounds
-    assert "-10000" not in out
+    assert out == expected
 
 
 def test_existing_float_constraints_preserved():
@@ -71,11 +63,9 @@ def test_existing_float_constraints_preserved():
 def t(y):
     pass
 """
+    expected = """from hypothesis import settings\nfrom hypothesis import given, strategies as st\n\n@given(y=st.floats(min_value=-1.0, max_value=1.0, allow_nan=False, allow_infinity=False))\n@settings(derandomize=True)\ndef t(y):\n    pass"""
     out = make_hypothesis_tests_deterministic(src)
-    assert "min_value=-1.0" in out or "min_value= -1.0" in out
-    assert "max_value=1.0" in out or "max_value= 1.0" in out
-    # Should not add the default 1e6 bounds
-    assert "1e6" not in out and "1000000" not in out
+    assert out == expected
 
 
 def test_idempotency():
@@ -99,14 +89,9 @@ def test_multiple_strategies_handled():
 def test_multi(a, b, c):
     pass
 """
+    expected = """from hypothesis import settings\nfrom hypothesis import given, strategies as st\n\n@given(a=st.integers(min_value=-10000, max_value=10000), b=st.integers(min_value=-10000, max_value=10000), c=st.floats(min_value=-1000000.0, max_value=1000000.0, allow_nan=False, allow_infinity=False))\n@settings(derandomize=True)\ndef test_multi(a, b, c):\n    pass"""
     out = make_hypothesis_tests_deterministic(src)
-    normalized = out.replace(" ", "").replace("\n", "")
-    # All integers should be constrained
-    assert normalized.count("min_value=-10000") >= 2
-    assert normalized.count("max_value=10000") >= 2
-    # Float should be constrained
-    assert "allow_nan=False" in normalized
-    assert "allow_infinity=False" in normalized
+    assert out == expected
 
 
 def test_settings_import_added_if_missing():
@@ -117,9 +102,9 @@ def test_settings_import_added_if_missing():
 def test_x(x):
     pass
 """
+    expected = """from hypothesis import settings\nfrom hypothesis import given, strategies as st\n\n@given(x=st.integers(min_value=-10000, max_value=10000))\n@settings(derandomize=True)\ndef test_x(x):\n    pass"""
     out = make_hypothesis_tests_deterministic(src)
-    # Should have settings import or settings in existing import
-    assert "settings" in out
+    assert out == expected
 
 
 def test_partial_constraints_completed():
@@ -130,11 +115,9 @@ def test_partial_constraints_completed():
 def test_x(x):
     pass
 """
+    expected = """from hypothesis import settings\nfrom hypothesis import given, strategies as st\n\n@given(x=st.integers(min_value=100))\n@settings(derandomize=True)\ndef test_x(x):\n    pass"""
     out = make_hypothesis_tests_deterministic(src)
-    # Should keep the min_value=100 and not override
-    assert "min_value=100" in out or "min_value= 100" in out
-    # Should not add default bounds since min_value exists
-    assert "-10000" not in out
+    assert out == expected
 
 
 def test_syntax_error_returns_original():
@@ -152,7 +135,6 @@ def test_no_hypothesis_code_unchanged():
 def test_regular():
     assert regular_function(2) == 4
 """
+    expected = """from hypothesis import settings\n\n@settings(derandomize=True)\ndef regular_function(x):\n    return x * 2\n\n@settings(derandomize=True)\ndef test_regular():\n    assert regular_function(2) == 4"""
     out = make_hypothesis_tests_deterministic(src)
-    # Should still parse and return valid code
-    assert "def regular_function" in out
-    assert "def test_regular" in out
+    assert out == expected