nifty->nifty2go

test/test_spaces/test_interface.py

@@ -24,7 +24,7 @@ from itertools import product
 from types import LambdaType
 from test.common import expand, generate_spaces, generate_harmonic_spaces
 
-from nifty.spaces import *
+from nifty2go.spaces import *
 
 
 class SpaceInterfaceTests(unittest.TestCase):

test/test_spaces/test_lm_space.py

@@ -22,7 +22,7 @@ import numpy as np
 
 from numpy.testing import assert_, assert_equal, assert_raises,\
         assert_almost_equal, assert_array_almost_equal
-from nifty import LMSpace
+from nifty2go import LMSpace
 from test.common import expand
 
 # [lmax, expected]

test/test_spaces/test_power_space.py

@@ -23,7 +23,7 @@ import numpy as np
 
 from numpy.testing import assert_, assert_equal, assert_almost_equal,\
         assert_raises
-from nifty import PowerSpace, RGSpace, Space, LMSpace
+from nifty2go import PowerSpace, RGSpace, Space, LMSpace
 from test.common import expand, marco_binbounds
 from itertools import product, chain
 

test/test_spaces/test_rg_space.py

@@ -23,7 +23,7 @@ import numpy as np
 
 from numpy.testing import assert_, assert_equal, assert_almost_equal, \
                           assert_array_equal
-from nifty import RGSpace
+from nifty2go import RGSpace
 from test.common import expand
 from itertools import product
 from nose.plugins.skip import SkipTest