whoops, fix my superficial test so it imports stuff it needs

trhowe · Feb 4, 2014 · 0fddc32 · 0fddc32
1 parent bf5f7cd
commit 0fddc32
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/contrib/stubs.py b/tests/contrib/stubs.py
@@ -1,5 +1,5 @@
 import unittest
-import datetime
+import time
 
 from nose.tools import assert_raises, assert_equal, assert_true
 
@@ -9,7 +9,7 @@
 from pycassa.contrib.stubs import ColumnFamilyStub, ConnectionPoolStub, \
                                   SystemManagerStub
 
-from pycassa.util import OrderedDict
+from pycassa.util import OrderedDict, convert_time_to_uuid
 
 pool = cf = None
 pool_stub = cf_stub = None