File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 1- # test_meem_engine.py
1+ # package/test/ test_meem_engine.py
22import pytest
33import numpy as np
44import sys
@@ -236,11 +236,14 @@ def test_build_problem_cache(sample_problem):
236236 assert np .any (cache .b_template != 0 )
237237
238238 # 4. Verify that the lists for m0-dependent parts are populated
239- assert len (cache .m0_dependent_A_indices ) > 0
239+ # [FIX]: Optimization replaced scalar entries (m0_dependent_A_indices)
240+ # with vectorized blocks (m0_dependent_blocks).
241+ assert len (cache .m0_dependent_blocks ) > 0
240242 assert len (cache .m0_dependent_b_indices ) > 0
241243
242244 # 5. Check a specific m0-dependent entry to ensure it's a callable
243- assert callable (cache .m0_dependent_A_indices [0 ][2 ])
245+ # m0_dependent_blocks format: (row_start, col_start, calc_func)
246+ assert callable (cache .m0_dependent_blocks [0 ][2 ])
244247 assert callable (cache .m0_dependent_b_indices [0 ][1 ])
245248
246249 print ("✅ Problem cache build test passed." )
Original file line number Diff line number Diff line change @@ -30,6 +30,7 @@ dependencies = [
3030 " pandas>=1.5" ,
3131 " matplotlib>=3.5" ,
3232 " h5netcdf>=0.12" ,
33+ " h5py>=3.0" ,
3334 " xarray>=2023.0" ,
3435 " streamlit>=1.0"
3536]
You can’t perform that action at this time.
0 commit comments