Add single-rank test for lsq_pseudoinv factory

Author: msimberg

model/common/tests/common/interpolation/mpi_tests/test_parallel_interpolation.py

@@ -232,8 +232,8 @@ def test_distributed_interpolation_lsq_pseudoinv(
     parallel_helpers.log_process_properties(processor_props)
     parallel_helpers.log_local_field_size(decomposition_info)
     factory = interpolation_factory_from_savepoint
-    field_ref_1 = interpolation_savepoint.__getattribute__("lsq_pseudoinv_1")().asnumpy()
-    field_ref_2 = interpolation_savepoint.__getattribute__("lsq_pseudoinv_2")().asnumpy()
+    field_ref_1 = interpolation_savepoint.lsq_pseudoinv_1().asnumpy()
+    field_ref_2 = interpolation_savepoint.lsq_pseudoinv_2().asnumpy()
     field_1 = factory.get(attrs.LSQ_PSEUDOINV)[:, 0, :]
     field_2 = factory.get(attrs.LSQ_PSEUDOINV)[:, 1, :]
     assert test_utils.dallclose(field_1, field_ref_1, atol=1e-15)  # type: ignore[arg-type] # mypy does not recognize sliced array as still an array

model/common/tests/common/interpolation/unit_tests/test_interpolation_factory.py

@@ -374,3 +374,20 @@ def test_rbf_interpolation_coeffs_vertex(
         field_v2[horizontal_start:],
         atol=RBF_TOLERANCES[dims.VertexDim][experiment.name],
     )
+
+
+@pytest.mark.level("integration")
+@pytest.mark.datatest
+def test_lsq_pseudoinv(
+    interpolation_savepoint: serialbox.InterpolationSavepoint,
+    experiment: definitions.Experiment,
+    backend: gtx_typing.Backend | None,
+) -> None:
+    field_ref_1 = interpolation_savepoint.lsq_pseudoinv_1()
+    field_ref_2 = interpolation_savepoint.lsq_pseudoinv_2()
+    factory = _get_interpolation_factory(backend, experiment)
+    field = factory.get(attrs.LSQ_PSEUDOINV).asnumpy()
+    field_1 = field[:, 0, :]
+    field_2 = field[:, 1, :]
+    assert test_helpers.dallclose(field_ref_1.asnumpy(), field_1, atol=1e-15)
+    assert test_helpers.dallclose(field_ref_2.asnumpy(), field_2, atol=1e-15)