test_context_must_be_str

oconnor663 · Jan 7, 2022 · ba13d28 · ba13d28
1 parent adee0a4
commit ba13d28
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/tests/test_blake3.py b/tests/test_blake3.py
@@ -348,3 +348,14 @@ def test_keyword_only_arguments():
 def test_usedforsecurity_ignored():
     blake3(usedforsecurity=True)
     blake3(usedforsecurity=False)
+
+
+def test_context_must_be_str():
+    # string works
+    blake3(derive_key_context="foo")
+    try:
+        # bytes fails
+        blake3(derive_key_context=b"foo")
+        assert False, "should fail"
+    except TypeError:
+        pass