Only allow "assertBytesEqual" for bytes.

exhuma · Oct 5, 2016 · 0c8dd57 · 0c8dd57
1 parent 7445e67
commit 0c8dd57
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/puresnmp/test/__init__.py b/puresnmp/test/__init__.py
@@ -13,6 +13,9 @@ def assertBytesEqual(self, a, b):
         """
         Helper method to compare bytes with more helpful output.
         """
+        if not isinstance(a, bytes) or not isinstance(b, bytes):
+            raise ValueError('assertBytesEqual requires two bytes objects!')
+
         if a != b:
             comparisons = []
             for offset, (char_a, char_b) in enumerate(zip_longest(a, b)):