honee/tests/smartsim_regression_framework.py

19 sys.path.insert(0, (Path(__file__).parents[3] / "tests/junit-xml").as_posix())
56 def verify_training_data(database_array, correct_array, ceed_resource, atol=1e-8, rtol=1e-8):
60     Check whether the S-frame-oriented vorticity vector's second component is just flipped.
68         idx_notclose = np.where(np.abs(database_array - correct_array) > total_tolerances)
75             test_fail = False if np.allclose(-database_vorticity, correct_vorticity,
82 …rror(f"Array values in database max difference: {np.max(np.abs(correct_array - database_array))}\n"
113     def test(self, ceed_resource) -> Tuple[bool, Exception, str]:
119                 '-ceed', ceed_resource,
120                 '-options_file', (file_dir / '../examples/blasius.yaml').as_posix(),
121                 '-ts_max_steps', '2',
122                 '-diff_filter_grid_based_width',
123                 '-ts_monitor_wall_clock_time', '-snes_monitor', '-ts_view_pre',
124                 '-diff_filter_ksp_max_it', '50', '-diff_filter_ksp_monitor',
125                 '-degree', '1',
126                 '-sgs_train_enable',
127                 '-sgs_train_write_data_interval', '2',
128                 '-sgs_train_filter_width_scales', '1.2,3.1',
129                 '-bc_symmetry_z',
130                 '-dm_plex_shape', 'zbox',
131                 '-dm_plex_box_bd', 'none,none,periodic',
132                 '-dm_plex_box_faces', '4,6,1',
133                 '-meshtransform_type', '-ts_monitor_smartsim_solution',
141             self.exp.start(client_exp, summary=False, block=True)
148             assert client.poll_tensor("check-run", 250, 5)
149             assert_equal(client.get_tensor("check-run")[0], 1)
151             assert client.poll_tensor("tensor-ow", 250, 5)
152             assert_equal(client.get_tensor("tensor-ow")[0], 1)
205                              elapsed_sec=time.time() - start,
207                                  '%Y-%m-%d %H:%M:%S %Z', time.localtime(start)),
228         '-c',
229         '--ceed-backends',
237     print("Setting up database...", end='')
240     print(" Done!")
242         print("working on " + ceed_resource + ' ...', end='')
246             print("Passed!")
248             print("Failed!", file=sys.stderr)
249 …print('\t' + ''.join(traceback.TracebackException.from_exception(exception).format()), file=sys.st…
251     print("Cleaning up database...", end='')
253     print(" Done!")