Switch to verbose logging output in mpi_array.tests.

mpi-array · Jun 4, 2017 · 8be8d11 · 8be8d11
1 parent 4384ec1
commit 8be8d11
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/mpi_array/local_test.py b/mpi_array/local_test.py
@@ -261,6 +261,7 @@ def test_views_2d(self):
         lary = mpi_array.local.ones(decomp=decomp, dtype="int64")
         self.assertEqual(_np.dtype("int64"), lary.dtype)
         rank_logger = _logging.get_rank_logger(self.id(), comm=decomp.rank_comm)
+        rank_logger.info("========================================================")
         rank_logger.info("rank_view_slice_n = %s" % (lary.decomp.rank_view_slice_n,))
         rank_logger.info("rank_view_slice_h = %s" % (lary.decomp.rank_view_slice_h,))
         rank_logger.info(
@@ -275,6 +276,7 @@ def test_views_2d(self):
             lary.rank_view_h[lary.decomp.rank_view_relative_slice_n].tolist(),
             lary.rank_view_n.tolist()
         )
+        rank_logger.info("^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^")
 
 
 _unittest.main(__name__)

diff --git a/mpi_array/tests.py b/mpi_array/tests.py
@@ -14,6 +14,7 @@
 from .license import license as _license, copyright as _copyright
 import pkg_resources as _pkg_resources
 import unittest as _unittest
+from mpi_array.unittest import main as _unittest_main
 import doctest as _doctest  # noqa: E402,F401
 
 import os.path
@@ -60,4 +61,4 @@ def load_tests(loader, tests, pattern):
 __all__ = [s for s in dir() if not s.startswith('_')]
 
 if __name__ == "__main__":
-    _unittest.main()
+    _unittest_main(__name__)