Updated example to print out difference values between the two models

MintonGroup · Feb 27, 2024 · 6f7fb92 · 6f7fb92
1 parent e673113
commit 6f7fb92
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/examples/spherical_harmonics_cb/J2_test_tp.py b/examples/spherical_harmonics_cb/J2_test_tp.py
@@ -82,5 +82,8 @@
 sim_obl.add_body(name=name_tp, a=a_tp, e=e_tp, inc=inc_tp, capom=capom_tp, omega=omega_tp, capm=capm_tp)
 sim_obl.run(tstart=0.0, tstop=10.0, dt=0.01, istep_out=10, dump_cadence=0, compute_conservation_values=True)
 
-ds_diff = sim_shgrav.data - sim_obl.data
+diff_vars = ['a','e','inc','capom','omega','capm','rh','vh']
+ds_diff = sim_shgrav.data[diff_vars] - sim_obl.data[diff_vars]
+
+print(ds_diff.isel(time=-1,name=-1))