Missing pytest imports

Author: sbird

nbodykit/algorithms/tests/test_fiber_colls.py

@@ -3,7 +3,7 @@
 from nbodykit.utils import ScatterArray, GatherArray
 from numpy.testing import assert_array_equal
 from mpi4py import MPI
-
+import pytest
 # debug logging
 setup_logging("debug")
 

nbodykit/algorithms/tests/test_fof.py

@@ -1,7 +1,7 @@
 from nbodykit.lab import *
 from nbodykit import setup_logging
 from mpi4py import MPI
-
+import pytest
 from numpy.testing import assert_allclose, assert_array_equal
 
 # debug logging

nbodykit/algorithms/tests/test_kddensity.py

@@ -2,7 +2,7 @@
 from nbodykit import setup_logging
 from numpy.testing import assert_array_equal, assert_allclose
 from mpi4py import MPI
-
+import pytest
 from nbodykit.algorithms.kdtree import KDDensity
 
 # debug logging

nbodykit/algorithms/tests/test_threeptcf.py

@@ -3,6 +3,7 @@
 from numpy.testing import assert_allclose, assert_array_equal
 import os
 from mpi4py import MPI
+import pytest
 
 setup_logging("debug")
 

nbodykit/algorithms/tests/test_zhist.py

@@ -2,6 +2,7 @@
 from nbodykit import setup_logging
 from numpy.testing import assert_allclose, assert_array_equal
 from mpi4py import MPI
+import pytest
 # debug logging
 setup_logging("debug")
 

nbodykit/io/tests/test_bigfile.py

@@ -5,6 +5,7 @@
 import pickle
 import contextlib
 from mpi4py import MPI
+import pytest
 
 @contextlib.contextmanager
 def temporary_data():

nbodykit/source/catalog/tests/test_file.py

@@ -4,6 +4,7 @@
 import tempfile
 import os
 from mpi4py import MPI
+import pytest
 
 @pytest.mark.parametrize("comm", [MPI.COMM_WORLD,])
 @pytest.mark.mpi

nbodykit/source/catalog/tests/test_lognormal.py

@@ -1,6 +1,7 @@
 from nbodykit.lab import *
 from nbodykit import setup_logging
 from nbodykit.utils import GatherArray
+import pytest
 
 from numpy.testing import assert_allclose
 from numpy.testing import assert_array_equal

nbodykit/source/catalog/tests/test_uniform.py

@@ -3,6 +3,7 @@
 from nbodykit.utils import GatherArray
 from numpy.testing import assert_array_equal
 from mpi4py import MPI
+import pytest
 
 @pytest.mark.parametrize("comm", [MPI.COMM_WORLD,])
 @pytest.mark.mpi

nbodykit/source/mesh/tests/test_array.py

@@ -1,5 +1,6 @@
 from nbodykit.lab import *
 from nbodykit import setup_logging
+import pytest
 
 from numpy.testing import assert_allclose, assert_array_equal
 from mpi4py import MPI