honee/tests/smartsim_regression_framework.py

19 sys.path.insert(0, (Path(__file__).parents[3] / "tests/junit-xml").as_posix())
45         raise Exception(f"Expected {truth}, but got {test}") from e
53         raise Exception(f"Expected {truth}, but got {test}") from e
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,
80                 f"./y0_database_values_{ceed_resource.replace('/', '_')}.npy").absolute()
82 …raise AssertionError(f"Array values in database max difference: {np.max(np.abs(correct_array - dat…
83                                  f"Array saved to {database_output_path.as_posix()}")
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',
138 …      client_exp = self.exp.create_model(f"client_{ceed_resource.replace('/', '_')}", run_settings)
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)
204         test_case = TestCase(f'SmartSim Test {ceed_resource}',
205                              elapsed_sec=time.time() - start,
207                                  '%Y-%m-%d %H:%M:%S %Z', time.localtime(start)),
211                              category=f'SmartSim Tests')
228         '-c',
229         '--ceed-backends',