From ac3b90aa69e9ef065fcefca308ff0038db9009df Mon Sep 17 00:00:00 2001
From: Matthias Diener <mdiener@illinois.edu>
Date: Fri, 28 May 2021 14:38:37 -0500
Subject: [PATCH] doc and flake fixes

---
 arraycontext/__init__.py |  3 ++-
 arraycontext/pytest.py   | 16 +++++++++-------
 2 files changed, 11 insertions(+), 8 deletions(-)

diff --git a/arraycontext/__init__.py b/arraycontext/__init__.py
index 243fd03..35c313f 100644
--- a/arraycontext/__init__.py
+++ b/arraycontext/__init__.py
@@ -51,6 +51,7 @@ from .container.traversal import (
         from_numpy, to_numpy)
 
 from .impl.pyopencl import PyOpenCLArrayContext
+from .impl.pytato import PytatoArrayContext
 
 from .pytest import pytest_generate_tests_for_array_contexts
 
@@ -77,7 +78,7 @@ __all__ = (
         "thaw", "freeze",
         "from_numpy", "to_numpy",
 
-        "PyOpenCLArrayContext",
+        "PyOpenCLArrayContext", "PytatoArrayContext",
 
         "make_loopy_program",
 
diff --git a/arraycontext/pytest.py b/arraycontext/pytest.py
index 85387a6..5bad85c 100644
--- a/arraycontext/pytest.py
+++ b/arraycontext/pytest.py
@@ -1,6 +1,6 @@
 """
 .. currentmodule:: arraycontext
-.. autofunction:: pytest_generate_tests_for_pyopencl_array_context
+.. autofunction:: pytest_generate_tests_for_array_contexts
 """
 
 
@@ -42,8 +42,8 @@ class _PyOpenCLArrayContextFactory(_ContextFactory):
         return PyOpenCLArrayContext(cl.CommandQueue(ctx))
 
     def __str__(self):
-        return ("<PyOpenCL array context factory for <pyopencl.Device '%s' on '%s'>" %
-                (self.device.name.strip(),
+        return ("<PyOpenCL array context factory for <pyopencl.Device '%s' on '%s'>"
+                % (self.device.name.strip(),
                  self.device.platform.name.strip()))
 
 
@@ -54,8 +54,8 @@ class _PytatoArrayContextFactory(_ContextFactory):
         return PytatoArrayContext(cl.CommandQueue(ctx))
 
     def __str__(self):
-        return ("<Pytato array context factory for <pyopencl.Device '%s' on '%s'>" %
-                (self.device.name.strip(),
+        return ("<Pytato array context factory for <pyopencl.Device '%s' on '%s'>"
+                % (self.device.name.strip(),
                  self.device.platform.name.strip()))
 
 
@@ -97,8 +97,10 @@ def pytest_generate_tests_for_array_contexts(metafunc) -> None:
                     "'ctx_factory' / 'ctx_getter' as arguments.")
 
         for arg_dict in arg_values:
-            arg_dict["actx_factory"] = _PyOpenCLArrayContextFactory(arg_dict["device"])
-            arg_dict["actx_factory_pytato"] = _PytatoArrayContextFactory(arg_dict["device"])
+            arg_dict["actx_factory"] = _PyOpenCLArrayContextFactory(arg_dict[
+                                                                         "device"])
+            arg_dict["actx_factory_pytato"] = _PytatoArrayContextFactory(arg_dict[
+                                                                         "device"])
 
     arg_values_out = [
             tuple(arg_dict[name] for name in arg_names)
-- 
GitLab