diff options
Diffstat (limited to 'client/common_lib/test_unittest.py')
-rwxr-xr-x | client/common_lib/test_unittest.py | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/client/common_lib/test_unittest.py b/client/common_lib/test_unittest.py index 8f3ef9ff82..044d402eb3 100755 --- a/client/common_lib/test_unittest.py +++ b/client/common_lib/test_unittest.py @@ -1,4 +1,4 @@ -#!/usr/bin/python2 +#!/usr/bin/python3 #pylint: disable-msg=C0111 """Unit Tests for autotest.client.common_lib.test""" @@ -15,6 +15,7 @@ import common import mock as pymock import os import shutil +import six from six.moves import range from autotest_lib.client.common_lib import test @@ -127,7 +128,10 @@ class Test_base_test_execute(TestTestCase): self.test.postprocess.expect_call() self.test.process_failed_constraints.expect_call() - fake_time = iter(range(4)).next + if six.PY2: + fake_time = iter(range(4)).next + else: + fake_time = iter(range(4)).__next__ self.test.execute(iterations=1, test_length=3, _get_time=fake_time) self.god.check_playback() |