Allow num_regression to accept array-like inputs

src/pytest_regressions/num_regression.py

@@ -73,11 +73,21 @@ def check(
 
         __tracebackhide__ = True
 
+        for k, obj in data_dict.items():
+            if not isinstance(obj, np.ndarray):
+                try:
+                    arr = np.atleast_1d(np.asarray(obj))
+                except:
+                    raise
+                else:
+                    if np.issubdtype(arr.dtype, np.number):
+                        data_dict[k] = arr
+
         data_shapes = []
         for obj in data_dict.values():
             assert type(obj) in [
                 np.ndarray
-            ], "Only numpy arrays are valid for numeric_data_regression fixture.\n"
+            ], "Only objects that can be coerced to numpy arrays are valid for numeric_data_regression fixture.\n"
             shape = obj.shape
 
             assert len(shape) == 1, (

tests/test_num_regression.py

@@ -257,3 +257,39 @@ def test_arrays_of_same_size(num_regression):
         "world": np.zeros((1,), dtype=np.int),
     }
     num_regression.check(same_size_int_arrays)
+
+
+def test_simple_numbers(num_regression, data_regression):
+    data1 = 1.1
+    data2 = 2
+    num_regression.check({"data1": data1, "data2": data2})
+    data_regression.check({"data1": data1, "data2": data2})
+    data1 += 0.00000001
+    num_regression.check({"data1": data1, "data2": data2})  # passes, within tol
+    with pytest.raises(
+        AssertionError,
+        match="FILES DIFFER.*",
+    ):
+        data_regression.check({"data1": data1, "data2": data2})  # fails, must be exact
+
+
+def test_simple_list_of_numbers(num_regression):
+    data1 = [1.1, 1.1, 1.1]
+    data2 = [2, 2, 2]
+    num_regression.check({"data1": data1, "data2": data2})
+
+
+def test_simple_tuple_of_numbers(num_regression):
+    data1 = (1.1, 1.1, 1.1)
+    data2 = (2, 2, 2)
+    num_regression.check({"data1": data1, "data2": data2})
+
+
+def test_simple_list_of_mostly_numbers(num_regression):
+    data1 = [1.1, "not a number", 1.1]
+    data2 = [2, 2, 2]
+    with pytest.raises(
+        AssertionError,
+        match="Only objects that can be coerced to numpy arrays are valid for numeric_data_regression fixture.",
+    ):
+        num_regression.check({"data1": data1, "data2": data2})

tests/test_num_regression/test_simple_floats.csv

@@ -0,0 +1,2 @@
+,data1,data2
+0,1.1000000000000001,2.2000000000000002

tests/test_num_regression/test_simple_list_of_numbers.csv

@@ -0,0 +1,4 @@
+,data1,data2
+0,1.1000000000000001,2
+1,1.1000000000000001,2
+2,1.1000000000000001,2

tests/test_num_regression/test_simple_numbers.csv

@@ -0,0 +1,2 @@
+,data1,data2
+0,1.1000000000000001,2

tests/test_num_regression/test_simple_numbers.yml

@@ -0,0 +1,2 @@
+data1: 1.1
+data2: 2

tests/test_num_regression/test_simple_tuple_of_numbers.csv

@@ -0,0 +1,4 @@
+,data1,data2
+0,1.1000000000000001,2
+1,1.1000000000000001,2
+2,1.1000000000000001,2