Skip to content

Commit 8085d8a

Browse files
committed
update to verif module to fix sens dict bug.
1 parent 1074d0f commit 8085d8a

File tree

7 files changed

+55
-47
lines changed

7 files changed

+55
-47
lines changed

src/pyvale/verif/psens.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@
2121

2222

2323
def samp_times(sim_data: mh.SimData) -> dict[str, None | np.ndarray]:
24-
sim_dims = sens.get_sim_dims(sim_data)
24+
sim_dims = sens.SimTools.get_sim_dims(sim_data)
2525
sample_times = {}
2626

2727
sample_times["sim"] = None
@@ -107,7 +107,7 @@ def check_gold_measurements(sens_dict: dict[str,sens.SensorArrayPoint]) -> list[
107107
for ss in sens_dict:
108108
measurements = sens_dict[ss].calc_measurements()
109109

110-
load_path = psensconst.GOLD_PATH / f"{ss}.npy"
110+
load_path = psensconst.GOLD_PATH / f"{ss.lower()}.npy"
111111
if load_path.is_file():
112112
gold = np.load(load_path)
113113

src/pyvale/verif/psensmech.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@
88
import pyvale.sensorsim as sens
99
import pyvale.verif.psens as psens
1010
import pyvale.verif.psensconst as psensconst
11+
import pyvale.dataset as dataset
1112

1213
"""
1314
DEVELOPER VERIFICATION MODULE
@@ -21,7 +22,7 @@
2122
"""
2223

2324
def simdata_mech_2d() -> mh.SimData:
24-
data_path = sens.DataSet.mechanical_2d_path()
25+
data_path = dataset.mechanical_2d_path()
2526
sim_data = mh.ExodusReader(data_path).read_all_sim_data()
2627
sim_data = sens.scale_length_units(scale=1000.0,
2728
sim_data=sim_data,
@@ -30,7 +31,7 @@ def simdata_mech_2d() -> mh.SimData:
3031

3132

3233
def simdata_mech_3d() -> mh.SimData:
33-
data_path = sens.DataSet.element_case_output_path(sens.EElemTest.HEX20)
34+
data_path = dataset.element_case_output_path(dataset.EElemTest.HEX20)
3435
sim_data = mh.ExodusReader(data_path).read_all_sim_data()
3536
field_comps = ("disp_x","disp_y","disp_z")
3637
sim_data = sens.scale_length_units(scale=1000.0,
@@ -39,7 +40,7 @@ def simdata_mech_3d() -> mh.SimData:
3940
return sim_data
4041

4142
def sens_pos_2d() -> dict[str,np.ndarray]:
42-
sim_dims = sens.get_sim_dims(simdata_mech_2d())
43+
sim_dims = sens.SimTools.get_sim_dims(simdata_mech_2d())
4344
sens_pos = {}
4445

4546
x_lims = sim_dims["x"]

src/pyvale/verif/psensmultiphys.py

Lines changed: 8 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@
1313
import pyvale.verif.psensvector as psensvector
1414
import pyvale.verif.psenstensor as psenstensor
1515
import pyvale.verif.psensmech as psensmech
16+
import pyvale.dataset as dataset
1617

1718

1819
def load_simdata_list(data_paths: list[Path],
@@ -29,23 +30,23 @@ def load_simdata_list(data_paths: list[Path],
2930

3031

3132
def simdata_list_2d() -> list[mh.SimData]:
32-
data_paths = sens.DataSet.thermomechanical_2d_experiment_paths()
33+
data_paths = dataset.thermomechanical_2d_experiment_paths()
3334
disp_comps = ("disp_x","disp_y")
3435
return load_simdata_list(data_paths,disp_comps)
3536

3637

3738
def simdata_list_3d() -> list[mh.SimData]:
38-
data_paths = [sens.DataSet.element_case_output_path(sens.EElemTest.TET4),
39-
sens.DataSet.element_case_output_path(sens.EElemTest.TET10),
40-
sens.DataSet.element_case_output_path(sens.EElemTest.HEX8),
41-
sens.DataSet.element_case_output_path(sens.EElemTest.HEX20)]
39+
data_paths = [dataset.element_case_output_path(dataset.EElemTest.TET4),
40+
dataset.element_case_output_path(dataset.EElemTest.TET10),
41+
dataset.element_case_output_path(dataset.EElemTest.HEX8),
42+
dataset.element_case_output_path(dataset.EElemTest.HEX20)]
4243
disp_comps = ("disp_x","disp_y","disp_z")
4344
return load_simdata_list(data_paths,disp_comps)
4445

4546

4647
def sens_pos_2d() -> dict[str,np.ndarray]:
4748
# Geometry does not change
48-
sim_dims = sens.get_sim_dims(simdata_list_2d()[0])
49+
sim_dims = sens.SimTools.get_sim_dims(simdata_list_2d()[0])
4950
sens_pos = {}
5051

5152
x_lims = sim_dims["x"]
@@ -165,7 +166,7 @@ def exp_sim_2d() -> dict[str,sens.ExperimentSimulator]:
165166
# print(f"{ee=}")
166167
# print(80*"-")
167168
if err_chain_dict[ff][ee] is not None:
168-
err_int_opts = sens.ErrIntOpts()
169+
err_int_opts = sens.ErrIntOpts
169170
err_int = sens.ErrIntegrator(err_chain_dict[ff][ee],
170171
sens_data_dict[ss],
171172
this_sens.get_measurement_shape(),

src/pyvale/verif/psensscalar.py

Lines changed: 20 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -5,10 +5,12 @@
55
#===============================================================================
66
import copy
77
import numpy as np
8+
89
import pyvale.mooseherder as mh
910
import pyvale.sensorsim as sens
1011
import pyvale.verif.psens as psens
1112
import pyvale.verif.psensconst as psensconst
13+
import pyvale.dataset as dataset
1214

1315
"""
1416
DEVELOPER VERIFICATION MODULE
@@ -23,7 +25,7 @@
2325
# TODO: fix position locking for 3D field errors
2426

2527
def simdata_2d() -> mh.SimData:
26-
data_path = sens.DataSet.thermal_2d_path()
28+
data_path = dataset.thermal_2d_path()
2729
sim_data = mh.ExodusReader(data_path).read_all_sim_data()
2830
sim_data = sens.scale_length_units(scale=1000.0,
2931
sim_data=sim_data,
@@ -32,7 +34,7 @@ def simdata_2d() -> mh.SimData:
3234

3335

3436
def simdata_3d() -> mh.SimData:
35-
data_path = sens.DataSet.thermal_3d_path()
37+
data_path = dataset.thermal_3d_path()
3638
sim_data = mh.ExodusReader(data_path).read_all_sim_data()
3739
sim_data = sens.scale_length_units(scale=1000.0,
3840
sim_data=sim_data,
@@ -41,7 +43,7 @@ def simdata_3d() -> mh.SimData:
4143

4244

4345
def sens_pos_2d() -> dict[str,np.ndarray]:
44-
sim_dims = sens.get_sim_dims(simdata_2d())
46+
sim_dims = sens.SimTools.get_sim_dims(simdata_2d())
4547
sens_pos = {}
4648

4749
x_lims = sim_dims["x"]
@@ -58,7 +60,7 @@ def sens_pos_2d() -> dict[str,np.ndarray]:
5860

5961

6062
def sens_pos_3d() -> dict[str,np.ndarray]:
61-
sim_dims = sens.get_sim_dims(simdata_3d())
63+
sim_dims = sens.SimTools.get_sim_dims(simdata_3d())
6264

6365
sens_pos = {}
6466

@@ -254,11 +256,11 @@ def sens_noerrs(sim_data: mh.SimData,
254256

255257

256258
def sens_dict(sim_data: mh.SimData,
257-
sens_data_dict: dict[str,sens.SensorData],
259+
sens_data_dict: dict[str, sens.SensorData],
258260
tag: str
259-
) -> dict[str,sens.SensorArrayPoint]:
261+
) -> dict[str, sens.SensorArrayPoint]:
260262

261-
sens = {}
263+
sens_dict = {}
262264
for ss in sens_data_dict:
263265
sens_array = sens_noerrs(sim_data,
264266
sens_data_dict[ss],
@@ -276,18 +278,21 @@ def sens_dict(sim_data: mh.SimData,
276278
pos_lock[pos_lock_key])
277279

278280
for ee in err_chain_dict:
279-
tag = f"{tag}_{ss}_err-{ee}"
280-
sens[tag] = copy.deepcopy(sens_array)
281+
full_tag = f"{tag}_{ss}_err-{ee}"
282+
sens_dict[full_tag] = copy.deepcopy(sens_array)
281283

282284
if err_chain_dict[ee] is not None:
283285
err_int_opts = sens.ErrIntOpts()
284-
err_int = sens.ErrIntegrator(err_chain_dict[ee],
285-
sens_data_dict[ss],
286-
sens[tag].get_measurement_shape(),
287-
err_int_opts=err_int_opts)
288-
sens[tag].set_error_integrator(err_int)
286+
err_int = sens.ErrIntegrator(
287+
err_chain_dict[ee],
288+
sens_data_dict[ss],
289+
sens_dict[full_tag].get_measurement_shape(),
290+
err_int_opts=err_int_opts
291+
)
292+
sens_dict[full_tag].set_error_integrator(err_int)
293+
294+
return sens_dict
289295

290-
return sens
291296

292297

293298
def sens_2d_dict() -> dict[str,sens.SensorArrayPoint]:

src/pyvale/verif/psenstensor.py

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,7 @@ def sens_2d_dict() -> dict[str,sens.SensorArrayPoint]:
5959
sim_data = psensmech.simdata_mech_2d()
6060
sens_data_dict = psensmech.sens_data_2d_dict()
6161

62-
sens = {}
62+
sens_dict = {}
6363
for ss in sens_data_dict:
6464
sens_array = sens_2d_noerrs(sim_data,sens_data_dict[ss])
6565

@@ -76,24 +76,24 @@ def sens_2d_dict() -> dict[str,sens.SensorArrayPoint]:
7676

7777
for ee in err_chain_dict:
7878
tag = f"tens2d_{ss}_err-{ee}"
79-
sens[tag] = copy.deepcopy(sens_array)
79+
sens_dict[tag] = copy.deepcopy(sens_array)
8080

8181
if err_chain_dict[ee] is not None:
8282
err_int_opts = sens.ErrIntOpts()
8383
err_int = sens.ErrIntegrator(err_chain_dict[ee],
8484
sens_data_dict[ss],
85-
sens[tag].get_measurement_shape(),
85+
sens_dict[tag].get_measurement_shape(),
8686
err_int_opts=err_int_opts)
87-
sens[tag].set_error_integrator(err_int)
87+
sens_dict[tag].set_error_integrator(err_int)
8888

89-
return sens
89+
return sens_dict
9090

9191

9292
def sens_3d_dict() -> dict[str,sens.SensorArrayPoint]:
9393
sim_data = psensmech.simdata_mech_3d()
9494
sens_data_dict = psensmech.sens_data_3d_dict()
9595

96-
sens = {}
96+
sens_dict = {}
9797
for ss in sens_data_dict:
9898
sens_array = sens_3d_noerrs(sim_data,sens_data_dict[ss])
9999

@@ -110,14 +110,14 @@ def sens_3d_dict() -> dict[str,sens.SensorArrayPoint]:
110110

111111
for ee in err_chain_dict:
112112
tag = f"tens3d_{ss}_err-{ee}"
113-
sens[tag] = copy.deepcopy(sens_array)
113+
sens_dict[tag] = copy.deepcopy(sens_array)
114114

115115
if err_chain_dict[ee] is not None:
116116
err_int_opts = sens.ErrIntOpts()
117117
err_int = sens.ErrIntegrator(err_chain_dict[ee],
118118
sens_data_dict[ss],
119-
sens[tag].get_measurement_shape(),
119+
sens_dict[tag].get_measurement_shape(),
120120
err_int_opts=err_int_opts)
121-
sens[tag].set_error_integrator(err_int)
121+
sens_dict[tag].set_error_integrator(err_int)
122122

123-
return sens
123+
return sens_dict

src/pyvale/verif/psensvector.py

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@ def sens_2d_dict() -> dict[str,sens.SensorArrayPoint]:
5252
sim_data = psensmech.simdata_mech_2d()
5353
sens_data_dict = psensmech.sens_data_2d_dict()
5454

55-
sens = {}
55+
sens_dict = {}
5656
for ss in sens_data_dict:
5757
sens_array = sens_2d_noerrs(sim_data,sens_data_dict[ss])
5858

@@ -69,24 +69,24 @@ def sens_2d_dict() -> dict[str,sens.SensorArrayPoint]:
6969

7070
for ee in err_chain_dict:
7171
tag = f"vec2d_{ss}_err-{ee}"
72-
sens[tag] = copy.deepcopy(sens_array)
72+
sens_dict[tag] = copy.deepcopy(sens_array)
7373

7474
if err_chain_dict[ee] is not None:
7575
err_int_opts = sens.ErrIntOpts()
7676
err_int = sens.ErrIntegrator(err_chain_dict[ee],
7777
sens_data_dict[ss],
78-
sens[tag].get_measurement_shape(),
78+
sens_dict[tag].get_measurement_shape(),
7979
err_int_opts=err_int_opts)
80-
sens[tag].set_error_integrator(err_int)
80+
sens_dict[tag].set_error_integrator(err_int)
8181

82-
return sens
82+
return sens_dict
8383

8484

8585
def sens_3d_dict() -> dict[str,sens.SensorArrayPoint]:
8686
sim_data = psensmech.simdata_mech_3d()
8787
sens_data_dict = psensmech.sens_data_3d_dict()
8888

89-
sens = {}
89+
sens_dict = {}
9090
for ss in sens_data_dict:
9191
sens_array = sens_3d_noerrs(sim_data,sens_data_dict[ss])
9292

@@ -103,14 +103,14 @@ def sens_3d_dict() -> dict[str,sens.SensorArrayPoint]:
103103

104104
for ee in err_chain_dict:
105105
tag = f"vec3d_{ss}_err-{ee}"
106-
sens[tag] = copy.deepcopy(sens_array)
106+
sens_dict[tag] = copy.deepcopy(sens_array)
107107

108108
if err_chain_dict[ee] is not None:
109109
err_int_opts = sens.ErrIntOpts()
110110
err_int = sens.ErrIntegrator(err_chain_dict[ee],
111111
sens_data_dict[ss],
112-
sens[tag].get_measurement_shape(),
112+
sens_dict[tag].get_measurement_shape(),
113113
err_int_opts=err_int_opts)
114-
sens[tag].set_error_integrator(err_int)
114+
sens_dict[tag].set_error_integrator(err_int)
115115

116-
return sens
116+
return sens_dict

tests/pointsensors/test_psens.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -48,6 +48,7 @@
4848
)
4949
def test_gold_combined(get_sensors: Callable[[], Dict[str, Any]]) -> None:
5050
sensors = get_sensors()
51+
print(sensors)
5152
fails = psens.check_gold_measurements(sensors)
5253
assert not fails, "\n".join(fails)
5354

0 commit comments

Comments
 (0)