test NumpyArrayContext

Author: kaushikcfd

arraycontext/pytest.py

@@ -34,6 +34,7 @@
 
 from typing import Any, Callable, Dict, Sequence, Type, Union
 
+from arraycontext import NumpyArrayContext
 from arraycontext.context import ArrayContext
 
 
@@ -222,6 +223,26 @@ def __str__(self):
         return "<PytatoJAXArrayContext>"
 
 
+# {{{ _PytestArrayContextFactory
+
+class _NumpyArrayContextForTests(NumpyArrayContext):
+    def transform_loopy_program(self, t_unit):
+        return t_unit
+
+
+class _PytestNumpyArrayContextFactory(PytestArrayContextFactory):
+    def __init__(self, *args, **kwargs):
+        super().__init__()
+
+    def __call__(self):
+        return _NumpyArrayContextForTests()
+
+    def __str__(self):
+        return "<NumpyArrayContext>"
+
+# }}}
+
+
 _ARRAY_CONTEXT_FACTORY_REGISTRY: \
         Dict[str, Type[PytestArrayContextFactory]] = {
                 "pyopencl": _PytestPyOpenCLArrayContextFactoryWithClass,
@@ -230,6 +251,7 @@ def __str__(self):
                 "pytato:pyopencl": _PytestPytatoPyOpenCLArrayContextFactory,
                 "pytato:jax": _PytestPytatoJaxArrayContextFactory,
                 "eagerjax": _PytestEagerJaxArrayContextFactory,
+                "numpy": _PytestNumpyArrayContextFactory,
                 }
 
 

test/test_arraycontext.py

@@ -46,6 +46,7 @@
 )
 from arraycontext.pytest import (
     _PytestEagerJaxArrayContextFactory,
+    _PytestNumpyArrayContextFactory,
     _PytestPyOpenCLArrayContextFactoryWithClass,
     _PytestPytatoJaxArrayContextFactory,
     _PytestPytatoPyOpenCLArrayContextFactory,
@@ -97,6 +98,7 @@ class _PytatoPyOpenCLArrayContextForTestsFactory(
     _PytatoPyOpenCLArrayContextForTestsFactory,
     _PytestEagerJaxArrayContextFactory,
     _PytestPytatoJaxArrayContextFactory,
+    _PytestNumpyArrayContextFactory,
     ])