From b124c1efd060386e7d541d84d673b5c17c1495fb Mon Sep 17 00:00:00 2001
From: Matthias Diener <mdiener@illinois.edu>
Date: Mon, 14 Jun 2021 16:28:57 -0500
Subject: [PATCH] fix norm() again
---
arraycontext/fake_numpy.py | 3 ---
test/test_arraycontext.py | 3 +++
2 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/arraycontext/fake_numpy.py b/arraycontext/fake_numpy.py
index 04f66e0..28be07e 100644
--- a/arraycontext/fake_numpy.py
+++ b/arraycontext/fake_numpy.py
@@ -189,9 +189,6 @@ class BaseFakeNumpyLinalgNamespace:
for _, subary in serialize_container(ary)],
ord=ord)
- if len(ary.shape) != 1:
- raise NotImplementedError("only vector norms are implemented")
-
if ary.size == 0:
return 0
diff --git a/test/test_arraycontext.py b/test/test_arraycontext.py
index b632e20..19b427e 100644
--- a/test/test_arraycontext.py
+++ b/test/test_arraycontext.py
@@ -748,6 +748,9 @@ def test_norm_ord_none(actx_factory, ndim):
norm_a_ref = np.linalg.norm(a, ord=None)
norm_a = actx.np.linalg.norm(actx.from_numpy(a), ord=None)
+ if not np.isscalar(norm_a):
+ norm_a = actx.to_numpy(norm_a)
+
np.testing.assert_allclose(norm_a, norm_a_ref)
--
GitLab