diff --git a/.travis.yml b/.travis.yml index b6614b7..0e97c77 100644 --- a/.travis.yml +++ b/.travis.yml @@ -1,5 +1,6 @@ language: python python: - "2.7" + - "3.6" # command to run tests script: python test.py diff --git a/test.py b/test.py index 4b7bffb..05b35d4 100755 --- a/test.py +++ b/test.py @@ -5,7 +5,12 @@ import sys import os import unittest import codecs -from mock import patch, Mock, MagicMock + +if sys.version_info[0] < 3: + from mock import patch, Mock, MagicMock +else: + from unittest.mock import patch, Mock, MagicMock + mock_socket = MagicMock(name='zk.socket') sys.modules['zk.socket'] = mock_socket from zk import ZK, const diff --git a/zk/base.py b/zk/base.py index 738f005..969b3c3 100644 --- a/zk/base.py +++ b/zk/base.py @@ -5,11 +5,11 @@ from socket import AF_INET, SOCK_DGRAM, SOCK_STREAM, socket, timeout from struct import pack, unpack import codecs -from zk import const -from zk.attendance import Attendance -from zk.exception import ZKErrorConnection, ZKErrorResponse, ZKNetworkError -from zk.user import User -from zk.finger import Finger +from . import const +from .attendance import Attendance +from .exception import ZKErrorConnection, ZKErrorResponse, ZKNetworkError +from .user import User +from .finger import Finger def safe_cast(val, to_type, default=None): @@ -302,7 +302,7 @@ class ZK(object): def __reverse_hex(self, hex): data = '' - for i in reversed(xrange(len(hex) / 2)): + for i in reversed(range(len(hex) / 2)): data += hex[i * 2:(i * 2) + 2] return data