Skip to content

Commit 62df1e9

Browse files
committed
adding to existing test new apis
1 parent 3c39faa commit 62df1e9

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed

tests/test_cyclic_support.py

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -94,6 +94,9 @@ def test_cyc_support_from_model(cyclic_lin_rst):
9494
exp = cyc_support.expand_element_id(1, [0, 1, 2])
9595
assert np.allclose(exp.ids, [1, 10, 19])
9696

97+
cs = cyc_support.cs()
98+
assert np.allclose(cs.scoping.ids, [12])
99+
97100

98101
def test_cyc_support_from_to_operator(cyclic_lin_rst, server_type):
99102
data_sources = dpf.DataSources(cyclic_lin_rst, server=server_type)
@@ -184,6 +187,17 @@ def test_cyc_support_multistage(cyclic_multistage):
184187
)
185188
assert np.allclose(cyc_support.sectors_set_for_expansion(stage_num=1).ids, list(range(0, 12)))
186189

190+
high_low_map = my_cyclic_support.high_low_map()
191+
assert np.allclose(high_low_map.get_entity_data_by_id(1446), 1447)
192+
assert np.allclose(high_low_map.get_entity_data_by_id(2946), 2948)
193+
assert np.allclose(high_low_map.get_entity_data_by_id(1452), 1466)
194+
195+
low_high_map = my_cyclic_support.low_high_map(1)
196+
assert np.allclose(low_high_map.get_entity_data_by_id(995), 939)
197+
assert np.allclose(low_high_map.get_entity_data_by_id(53), 54)
198+
assert np.allclose(low_high_map.get_entity_data_by_id(70), 56)
199+
200+
187201

188202
def test_delete_cyc_support(cyclic_lin_rst, server_type_legacy_grpc):
189203
data_sources = dpf.DataSources(cyclic_lin_rst, server=server_type_legacy_grpc)

0 commit comments

Comments
 (0)