fix repr test

HypothesisWorks · Jan 10, 2024 · f4b6a87 · f4b6a87
1 parent 2453a60
commit f4b6a87
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/hypothesis-python/tests/cover/test_searchstrategy.py b/hypothesis-python/tests/cover/test_searchstrategy.py
@@ -13,6 +13,7 @@
 from collections import defaultdict, namedtuple
 
 import attr
+from hypothesis.internal.reflection import get_pretty_function_description
 import pytest
 
 from hypothesis.errors import InvalidArgument
@@ -77,12 +78,12 @@ def f(u, v):
 
 def test_can_map_nameless():
     f = nameless_const(2)
-    assert repr(f) in repr(integers().map(f))
+    assert get_pretty_function_description(f) in repr(integers().map(f))
 
 
 def test_can_flatmap_nameless():
     f = nameless_const(just(3))
-    assert repr(f) in repr(integers().flatmap(f))
+    assert get_pretty_function_description(f) in repr(integers().flatmap(f))
 
 
 def test_flatmap_with_invalid_expand():