fix 3.10 issue maybe

Author: Robbie Muir

test/test_common.py

@@ -16,6 +16,7 @@
 
 from linopy import LinearExpression, Variable
 from linopy.common import (
+    CoordAlignWarning,
     align,
     as_dataarray,
     assign_multiindex_safe,
@@ -98,7 +99,7 @@ def test_as_dataarray_with_series_override_coords() -> None:
     target_dim = "dim_0"
     target_index = ["a", "b", "c"]
     s = pd.Series([1, 2, 3], index=target_index)
-    with pytest.warns(UserWarning):
+    with pytest.warns(CoordAlignWarning):
         da = as_dataarray(s, coords=[[1, 2, 3]])
     assert isinstance(da, DataArray)
     assert da.dims == (target_dim,)
@@ -217,7 +218,7 @@ def test_as_dataarray_dataframe_override_coords() -> None:
     target_index = ["a", "b"]
     target_columns = ["A", "B"]
     df = pd.DataFrame([[1, 2], [3, 4]], index=target_index, columns=target_columns)
-    with pytest.warns(UserWarning):
+    with pytest.warns(CoordAlignWarning):
         da = as_dataarray(df, coords=[[1, 2], [2, 3]])
     assert isinstance(da, DataArray)
     assert da.dims == target_dims

test/test_variables.py

@@ -3,7 +3,6 @@
 This module aims at testing the correct behavior of the Variables class.
 """
 
-import sys
 import warnings
 from datetime import datetime
 
@@ -142,14 +141,10 @@ def test_timezone_alignment_with_multiplication() -> None:
     series1 = pd.Series(index=utc_index, data=1.0)
     var1 = model.add_variables(coords=[utc_index], name="var1")
 
-    if sys.version_info >= (3, 11):
-        with warnings.catch_warnings(category=CoordAlignWarning):
-            warnings.simplefilter("error")
-            expr = var1 * series1
-    else:
-        with warnings.catch_warnings():
-            warnings.simplefilter("error")
-            expr = var1 * series1
+    # TODO increase coverage for datarray when coords are not dataarray
+    with warnings.catch_warnings():
+        warnings.simplefilter("error", CoordAlignWarning)
+        expr = var1 * series1
 
     index: pd.DatetimeIndex = expr.coords["time"].to_index()
     assert index.equals(utc_index)