Fixing the test dependencies

kayak · Aug 17, 2016 · 9db073b · 9db073b
1 parent 7771625
commit 9db073b
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 6 deletions.
diff --git a/fireant/slicer/transformers/datatables.py b/fireant/slicer/transformers/datatables.py
@@ -1,8 +1,4 @@
 # coding: utf-8
-
-# This import is necessary since Pandas uses it in some places
-from __future__ import unicode_literals
-
 import numpy as np
 import pandas as pd
 from builtins import str

diff --git a/fireant/tests/slicer/transformers/test_datatables.py b/fireant/tests/slicer/transformers/test_datatables.py
@@ -428,7 +428,7 @@ def test_nan_data_point(self):
 
     def test_str_data_point(self):
         result = datatables._format_data_point(u'abc')
-        self.assertEqual(u'abc', result)
+        self.assertEqual('abc', result)
 
     def test_int64_data_point(self):
         # Needs to be cast to python int

diff --git a/setup.py b/setup.py
@@ -58,7 +58,7 @@ def readme():
         'pypika==0.0.26'
     ],
     tests_require=[
-        'mock',
+        'mock', 'vertica-python>=0.6'
     ],
     extras_require={
         'vertica': ['vertica-python>=0.6'],