diff --git a/tests/finance/test_slippage.py b/tests/finance/test_slippage.py index b5ad0380..44f1fb8a 100644 --- a/tests/finance/test_slippage.py +++ b/tests/finance/test_slippage.py @@ -20,7 +20,7 @@ import datetime import pytz -from unittest2 import TestCase +from unittest import TestCase from zipline.finance.slippage import VolumeShareSlippage diff --git a/tests/test_algorithm.py b/tests/test_algorithm.py index 0f761fa8..048e50b8 100644 --- a/tests/test_algorithm.py +++ b/tests/test_algorithm.py @@ -13,7 +13,7 @@ # See the License for the specific language governing permissions and # limitations under the License. -from unittest2 import TestCase +from unittest import TestCase from datetime import timedelta import numpy as np diff --git a/tests/test_exception_handling.py b/tests/test_exception_handling.py index cfd9e6b3..be226d81 100644 --- a/tests/test_exception_handling.py +++ b/tests/test_exception_handling.py @@ -13,7 +13,7 @@ # See the License for the specific language governing permissions and # limitations under the License. -from unittest2 import TestCase +from unittest import TestCase import zipline.utils.simfactory as simfactory from zipline.test_algorithms import ( diff --git a/tests/test_finance.py b/tests/test_finance.py index a30d492f..28a66f59 100644 --- a/tests/test_finance.py +++ b/tests/test_finance.py @@ -18,7 +18,7 @@ Tests for the zipline.finance package """ import pytz -from unittest2 import TestCase +from unittest import TestCase from datetime import datetime, timedelta from nose.tools import timed diff --git a/tests/test_sorting.py b/tests/test_sorting.py index 3903a7a5..fb836c4c 100644 --- a/tests/test_sorting.py +++ b/tests/test_sorting.py @@ -15,7 +15,7 @@ import pytz -from unittest2 import TestCase +from unittest import TestCase from itertools import chain, izip_longest from datetime import datetime, timedelta from collections import deque diff --git a/tests/test_sources.py b/tests/test_sources.py index 39decc41..1257ac55 100644 --- a/tests/test_sources.py +++ b/tests/test_sources.py @@ -13,7 +13,7 @@ # See the License for the specific language governing permissions and # limitations under the License. -from unittest2 import TestCase +from unittest import TestCase import zipline.utils.factory as factory from zipline.sources import DataFrameSource diff --git a/tests/test_transforms.py b/tests/test_transforms.py index 3ad29435..8c979095 100644 --- a/tests/test_transforms.py +++ b/tests/test_transforms.py @@ -18,7 +18,7 @@ import numpy as np import pandas as pd from datetime import timedelta, datetime -from unittest2 import TestCase +from unittest import TestCase from zipline import ndict diff --git a/tests/test_utils.py b/tests/test_utils.py index b5fafa7d..00385dbc 100644 --- a/tests/test_utils.py +++ b/tests/test_utils.py @@ -13,7 +13,7 @@ # See the License for the specific language governing permissions and # limitations under the License. -from unittest2 import TestCase +from unittest import TestCase from zipline.utils.factory import load_from_yahoo import pandas as pd import pytz