Fix Python 2.6

This commit is contained in:
Ben Firshman 2014-01-06 11:22:46 +00:00
parent 8de07ccf65
commit f96a1a0b35
2 changed files with 9 additions and 2 deletions

View File

@ -0,0 +1,7 @@
import sys
if sys.version_info >= (2,7):
import unittest
else:
import unittest2 as unittest

View File

@ -1,10 +1,10 @@
from docker import Client from docker import Client
from fig.service import Service from fig.service import Service
from fig.cli.utils import docker_url from fig.cli.utils import docker_url
from unittest2 import TestCase from . import unittest
class DockerClientTestCase(TestCase): class DockerClientTestCase(unittest.TestCase):
@classmethod @classmethod
def setUpClass(cls): def setUpClass(cls):
cls.client = Client(docker_url()) cls.client = Client(docker_url())