use deriv_diff_map in test

IDAES · Jan 7, 2022 · c3d314b · c3d314b
1 parent 2a99ff6
commit c3d314b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/idaes/core/solvers/tests/test_petsc.py b/idaes/core/solvers/tests/test_petsc.py
@@ -79,7 +79,8 @@ def test_set_dae_suffix():
     constraints = [con[t] for con in time_cons if t in con]
     variables = [var[t] for var in time_vars]
     t_block = create_subsystem_block(constraints, variables)
-    petsc._set_dae_suffixes_from_variables(t_block, variables)
+    deriv_diff_map =  petsc._get_derivative_differential_data_map(m, m.t)
+    petsc._set_dae_suffixes_from_variables(t_block, variables, deriv_diff_map)
     petsc._sub_problem_scaling_suffix(m, t_block)
 
     assert t_block.dae_suffix[m.ydot[180, 1]] == 2