summaryrefslogtreecommitdiff
path: root/tests
diff options
context:
space:
mode:
Diffstat (limited to 'tests')
-rw-r--r--tests/test_bcrypt.py5
1 files changed, 3 insertions, 2 deletions
diff --git a/tests/test_bcrypt.py b/tests/test_bcrypt.py
index 920facf..7f589fb 100644
--- a/tests/test_bcrypt.py
+++ b/tests/test_bcrypt.py
@@ -2,6 +2,7 @@ import os
import mock
import pytest
+import six
import bcrypt
@@ -109,9 +110,9 @@ def test_hashpw_invalid():
def test_hashpw_str_password():
with pytest.raises(TypeError):
- bcrypt.hashpw(bcrypt.text_type("password"), b"$2a$04$cVWp4XaNU8a4v1uMRum2SO")
+ bcrypt.hashpw(six.text_type("password"), b"$2a$04$cVWp4XaNU8a4v1uMRum2SO")
def test_hashpw_str_salt():
with pytest.raises(TypeError):
- bcrypt.hashpw(b"password", bcrypt.text_type("$2a$04$cVWp4XaNU8a4v1uMRum2SO"))
+ bcrypt.hashpw(b"password", six.text_type("$2a$04$cVWp4XaNU8a4v1uMRum2SO"))