Update test_validation.py

HypothesisWorks · Sep 9, 2019 · 6fd32b0 · 6fd32b0
1 parent 7177a65
commit 6fd32b0
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/hypothesis-python/src/hypothesis/internal/coverage.py b/hypothesis-python/src/hypothesis/internal/coverage.py
@@ -50,7 +50,8 @@ def pretty_file_name(f):
         pass
 
     parts = f.split(os.path.sep)
-    parts = parts[-parts[::-1].index("hypothesis") :]
+    if "hypothesis" in parts:
+        parts = parts[-parts[::-1].index("hypothesis") :]
     result = os.path.sep.join(parts)
     pretty_file_name_cache[f] = result
     return result

diff --git a/hypothesis-python/tests/cover/test_validation.py b/hypothesis-python/tests/cover/test_validation.py
@@ -253,8 +253,10 @@ def test(x):
 def test_check_type_with_tuple_of_length_two():
     # This test covers logic for length-two tuples that is essential on PY2,
     # e.g. string_types (str, unicode) which are all length-one on Python 3.
-    check_type((int, str), 1)
-    check_type((int, str), "1")
+    def type_checker(x):
+        check_type((int, str), x, "x")
 
-    with pytest.raises(InvalidArgument):
-        check_type((int, str), 1.0)
+    type_checker(1)
+    type_checker("1")
+    with pytest.raises(InvalidArgument, match="Expected one of int, str but got "):
+        type_checker(1.0)