Fixed linear_modeling tests under python3

berkeley-stat159 · Dec 8, 2015 · dcd399f · dcd399f
1 parent 509e2ce
commit dcd399f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/code/utils/linear_modeling.py b/code/utils/linear_modeling.py
@@ -5,7 +5,7 @@
 import numpy.linalg as npl
 from scipy.stats import t as t_dist
 from scipy.ndimage import gaussian_filter
-import scene_slicer as ssm
+from . import scene_slicer as ssm
 import nibabel as nib
 
 

diff --git a/code/utils/tests/test_linear_modeling.py b/code/utils/tests/test_linear_modeling.py
@@ -68,7 +68,7 @@ def test_get_top_32():
 
 def test_get_index_4d():
     actual = linear_modeling.get_index_4d([0, 2], data)
-    assert_equal(actual, [(0, 0), (0, 2)])
+    assert_equal(list(actual), [(0, 0), (0, 2)])
 
     # def test_plot_single_voxel():
     # 	not sure how to test plots