Switch to using six

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"))