fix test

Author: Robbie Muir

linopy/common.py

@@ -42,6 +42,9 @@
     from linopy.variables import Variable
 
 
+class CoordAlignWarning(UserWarning): ...
+
+
 def set_int_index(series: pd.Series) -> pd.Series:
     """
     Convert string index to int index.
@@ -189,7 +192,8 @@ def pandas_to_dataarray(
                 f"coords for dimension(s) {non_aligned} is not aligned with the pandas object. "
                 "Previously, the indexes of the pandas were ignored and overwritten in "
                 "these cases. Now, the pandas object's coordinates are taken considered"
-                " for alignment."
+                " for alignment.",
+                CoordAlignWarning,
             )
 
     return DataArray(arr, coords=None, dims=dims, **kwargs)
@@ -469,7 +473,7 @@ def save_join(*dataarrays: DataArray, integer_dtype: bool = False) -> Dataset:
     except ValueError:
         warn(
             "Coordinates across variables not equal. Perform outer join.",
-            UserWarning,
+            CoordAlignWarning,
         )
         arrs = xr_align(*dataarrays, join="outer")
         if integer_dtype:

test/test_variables.py

@@ -4,17 +4,19 @@
 """
 
 import warnings
-from datetime import UTC, datetime
+from datetime import datetime
 
 import numpy as np
 import pandas as pd
 import pytest
 import xarray as xr
 import xarray.core.indexes
 import xarray.core.utils
+from pytz import UTC
 
 import linopy
 from linopy import Model
+from linopy.common import CoordAlignWarning
 from linopy.testing import assert_varequal
 from linopy.variables import ScalarVariable
 
@@ -139,9 +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")
 
-    with warnings.catch_warnings():
+    with warnings.catch_warnings(category=CoordAlignWarning):
         warnings.simplefilter("error")
         expr = var1 * series1
+
     index: pd.DatetimeIndex = expr.coords["time"].to_index()  # type: ignore
     assert index.equals(utc_index)
     assert index.tzinfo is UTC