diff options
| -rw-r--r-- | tests/hazmat/primitives/twofactor/test_hotp.py | 4 | ||||
| -rw-r--r-- | tests/hazmat/primitives/twofactor/test_totp.py | 4 |
2 files changed, 2 insertions, 6 deletions
diff --git a/tests/hazmat/primitives/twofactor/test_hotp.py b/tests/hazmat/primitives/twofactor/test_hotp.py index 0ef18979..548c6264 100644 --- a/tests/hazmat/primitives/twofactor/test_hotp.py +++ b/tests/hazmat/primitives/twofactor/test_hotp.py @@ -15,8 +15,6 @@ from __future__ import absolute_import, division, print_function import os -import pretend - import pytest from cryptography.exceptions import InvalidToken, UnsupportedInterface @@ -102,7 +100,7 @@ class TestHOTP(object): def test_invalid_backend(): secret = b"12345678901234567890" - pretend_backend = pretend.stub() + pretend_backend = object() with pytest.raises(UnsupportedInterface): HOTP(secret, 8, hashes.SHA1(), pretend_backend) diff --git a/tests/hazmat/primitives/twofactor/test_totp.py b/tests/hazmat/primitives/twofactor/test_totp.py index 776a9f2a..294c19ab 100644 --- a/tests/hazmat/primitives/twofactor/test_totp.py +++ b/tests/hazmat/primitives/twofactor/test_totp.py @@ -13,8 +13,6 @@ from __future__ import absolute_import, division, print_function -import pretend - import pytest from cryptography.exceptions import InvalidToken, UnsupportedInterface @@ -136,7 +134,7 @@ class TestTOTP(object): def test_invalid_backend(): secret = b"12345678901234567890" - pretend_backend = pretend.stub() + pretend_backend = object() with pytest.raises(UnsupportedInterface): TOTP(secret, 8, hashes.SHA1(), 30, pretend_backend) |