test: update test to match new Rotator structure

xarray-contrib · Sep 13, 2024 · cce4e35 · cce4e35
1 parent af1ca8f
commit cce4e35
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/models/single/test_hilbert_eof_rotator.py b/tests/models/single/test_hilbert_eof_rotator.py
@@ -42,12 +42,12 @@ def test_fit(ceof_model):
  ceof_rotator.fit(ceof_model)
 
  assert hasattr(
- ceof_rotator, "model"
- ), 'The attribute "model" should be populated after fitting.'
+ ceof_rotator, "model_data"
+ ), 'The attribute "model_data" should be populated after fitting.'
  assert hasattr(
  ceof_rotator, "data"
  ), 'The attribute "data" should be populated after fitting.'
- assert isinstance(ceof_rotator.model, HilbertEOF)
+ assert isinstance(ceof_rotator.model_data, DataContainer)
  assert isinstance(ceof_rotator.data, DataContainer)