bugfix: Fixed multiplication with numpy constants (#494)

Author: RobbieKiwi

doc/release_notes.rst

@@ -4,6 +4,7 @@ Release Notes
 .. Upcoming Version
 .. ----------------
 .. * Improved constraint equality check in `linopy.testing.assert_conequal` to less strict optionally
+.. * Minor bugfix for multiplying variables with numpy type constants
 
 Version 0.5.6
 --------------

linopy/common.py

@@ -254,7 +254,9 @@ def as_dataarray(
         arr = pandas_to_dataarray(arr, coords=coords, dims=dims, **kwargs)
     elif isinstance(arr, np.ndarray):
         arr = numpy_to_dataarray(arr, coords=coords, dims=dims, **kwargs)
-    elif isinstance(arr, np.number | int | float | str | bool | list):
+    elif isinstance(arr, np.number):
+        arr = DataArray(float(arr), coords=coords, dims=dims, **kwargs)
+    elif isinstance(arr, int | float | str | bool | list):
         arr = DataArray(arr, coords=coords, dims=dims, **kwargs)
 
     elif not isinstance(arr, DataArray):

test/test_common.py

@@ -369,6 +369,14 @@ def test_as_dataarray_with_number() -> None:
     assert list(da.coords["dim1"].values) == ["a"]
 
 
+def test_as_dataarray_with_np_number() -> None:
+    num = np.float64(1)
+    da = as_dataarray(num, dims=["dim1"], coords=[["a"]])
+    assert isinstance(da, DataArray)
+    assert da.dims == ("dim1",)
+    assert list(da.coords["dim1"].values) == ["a"]
+
+
 def test_as_dataarray_with_number_default_dims_coords() -> None:
     num = 1
     da = as_dataarray(num)