Skip to content
Snippets Groups Projects
Commit 324d5151 authored by Michal Maciejewski's avatar Michal Maciejewski
Browse files

Update test/test_reference/test_Reference.py, lhcsmapi/__init__.py files

parent 1d7ec18d
No related branches found
No related tags found
No related merge requests found
Pipeline #2288411 passed with warnings
name = "lhcsmapi"
__version__ = "1.4.78"
__version__ = "1.4.79"
def get_lhcsmapi_version() -> None:
......
......@@ -181,331 +181,7 @@ class TestReference(unittest.TestCase):
# assert
period_exp = 'eYETS 2017/2018'
self.assertEqual(period_exp, period_act)
# Nominal voltage-dependent QH discharge reference criteria - 900 V
def test_read_quench_heaters_reference_features_rb(self):
# arrange
circuit_type = 'RB'
# act
features_act = Reference.read_quench_heaters_reference_features(circuit_type)
# assert
features_ref = pd.DataFrame({'min': {'%CELL%:U_HDS_1:first': 780, '%CELL%:U_HDS_2:first': 780,
'%CELL%:U_HDS_3:first': 780, '%CELL%:U_HDS_4:first': 780,
'%CELL%:U_HDS_1:last20mean': 15, '%CELL%:U_HDS_2:last20mean': 15,
'%CELL%:U_HDS_3:last20mean': 15, '%CELL%:U_HDS_4:last20mean': 15},
'max': {'%CELL%:U_HDS_1:first': 980, '%CELL%:U_HDS_2:first': 980,
'%CELL%:U_HDS_3:first': 980, '%CELL%:U_HDS_4:first': 980,
'%CELL%:U_HDS_1:last20mean': 70, '%CELL%:U_HDS_2:last20mean': 70,
'%CELL%:U_HDS_3:last20mean': 70, '%CELL%:U_HDS_4:last20mean': 70}})
pd.testing.assert_frame_equal(features_ref, features_act)
def test_read_quench_heaters_reference_features_rq(self):
# arrange
circuit_type = 'RQ'
# act
features_act = Reference.read_quench_heaters_reference_features(circuit_type)
# assert
features_ref = pd.DataFrame({'min': {'%CELL%:U_HDS_1:first': 810, '%CELL%:U_HDS_2:first': 810,
'%CELL%:U_HDS_1:last20mean': 0, '%CELL%:U_HDS_2:last20mean': 0},
'max': {'%CELL%:U_HDS_1:first': 1020, '%CELL%:U_HDS_2:first': 1020,
'%CELL%:U_HDS_1:last20mean': 10, '%CELL%:U_HDS_2:last20mean': 10}})
pd.testing.assert_frame_equal(features_ref, features_act)
def test_read_quench_heaters_reference_features_it(self):
# arrange
circuit_type = 'IT'
# act
features_act = Reference.read_quench_heaters_reference_features(circuit_type)
# assert
features_ref = pd.DataFrame({'min': {'heat.%CIRCUIT%:U_HDS_1_Q1:first': 810,
'heat.%CIRCUIT%:U_HDS_2_Q1:first': 810,
'heat.%CIRCUIT%:U_HDS_1_Q2:first': 810,
'heat.%CIRCUIT%:U_HDS_2_Q2:first': 810,
'heat.%CIRCUIT%:U_HDS_3_Q2:first': 810,
'heat.%CIRCUIT%:U_HDS_4_Q2:first': 810,
'heat.%CIRCUIT%:U_HDS_1_Q3:first': 810,
'heat.%CIRCUIT%:U_HDS_2_Q3:first': 810,
'heat.%CIRCUIT%:U_HDS_1_Q1:last20mean': 0,
'heat.%CIRCUIT%:U_HDS_2_Q1:last20mean': 0,
'heat.%CIRCUIT%:U_HDS_1_Q2:last20mean': 0,
'heat.%CIRCUIT%:U_HDS_2_Q2:last20mean': 0,
'heat.%CIRCUIT%:U_HDS_3_Q2:last20mean': 0,
'heat.%CIRCUIT%:U_HDS_4_Q2:last20mean': 0,
'heat.%CIRCUIT%:U_HDS_1_Q3:last20mean': 0,
'heat.%CIRCUIT%:U_HDS_2_Q3:last20mean': 0},
'max': {'heat.%CIRCUIT%:U_HDS_1_Q1:first': 1000,
'heat.%CIRCUIT%:U_HDS_2_Q1:first': 1000,
'heat.%CIRCUIT%:U_HDS_1_Q2:first': 1000,
'heat.%CIRCUIT%:U_HDS_2_Q2:first': 1000,
'heat.%CIRCUIT%:U_HDS_3_Q2:first': 1000,
'heat.%CIRCUIT%:U_HDS_4_Q2:first': 1000,
'heat.%CIRCUIT%:U_HDS_1_Q3:first': 1000,
'heat.%CIRCUIT%:U_HDS_2_Q3:first': 1000,
'heat.%CIRCUIT%:U_HDS_1_Q1:last20mean': 10,
'heat.%CIRCUIT%:U_HDS_2_Q1:last20mean': 10,
'heat.%CIRCUIT%:U_HDS_1_Q2:last20mean': 10,
'heat.%CIRCUIT%:U_HDS_2_Q2:last20mean': 10,
'heat.%CIRCUIT%:U_HDS_3_Q2:last20mean': 10,
'heat.%CIRCUIT%:U_HDS_4_Q2:last20mean': 10,
'heat.%CIRCUIT%:U_HDS_1_Q3:last20mean': 10,
'heat.%CIRCUIT%:U_HDS_2_Q3:last20mean': 10}})
pd.testing.assert_frame_equal(features_ref, features_act)
def test_read_quench_heaters_reference_features_ipd2(self):
# arrange
circuit_type = 'IPD2'
# act
features_act = Reference.read_quench_heaters_reference_features(circuit_type)
# assert
features_ref = pd.DataFrame({'min': {'heat.%CIRCUIT%:U_HDS_1:first': 810, 'heat.%CIRCUIT%:U_HDS_2:first': 810,
'heat.%CIRCUIT%:U_HDS_1:last20mean': 0,
'heat.%CIRCUIT%:U_HDS_2:last20mean': 0},
'max': {'heat.%CIRCUIT%:U_HDS_1:first': 1000, 'heat.%CIRCUIT%:U_HDS_2:first': 1000,
'heat.%CIRCUIT%:U_HDS_1:last20mean': 10,
'heat.%CIRCUIT%:U_HDS_2:last20mean': 10}})
pd.testing.assert_frame_equal(features_ref, features_act)
def test_read_quench_heaters_reference_features_ipd2_b1b2(self):
# arrange
circuit_type = 'IPD2_B1B2'
# act
features_act = Reference.read_quench_heaters_reference_features(circuit_type)
# assert
features_ref = pd.DataFrame({'min': {'heat.%CIRCUIT%:U_HDS_1_B1:first': 810,
'heat.%CIRCUIT%:U_HDS_1_B2:first': 810,
'heat.%CIRCUIT%:U_HDS_1_B1:last20mean': 0,
'heat.%CIRCUIT%:U_HDS_1_B2:last20mean': 0},
'max': {'heat.%CIRCUIT%:U_HDS_1_B1:first': 1000,
'heat.%CIRCUIT%:U_HDS_1_B2:first': 1000,
'heat.%CIRCUIT%:U_HDS_1_B1:last20mean': 10,
'heat.%CIRCUIT%:U_HDS_1_B2:last20mean': 10}})
pd.testing.assert_frame_equal(features_ref, features_act)
def test_read_quench_heaters_reference_features_ipq2(self):
# arrange
circuit_type = 'IPQ2'
# act
features_act = Reference.read_quench_heaters_reference_features(circuit_type)
# assert
features_ref = pd.DataFrame({'min': {'heat.%CIRCUIT%:U_HDS_1_B1:first': 810,
'heat.%CIRCUIT%:U_HDS_1_B2:first': 810,
'heat.%CIRCUIT%:U_HDS_1_B1:last20mean': 0,
'heat.%CIRCUIT%:U_HDS_1_B2:last20mean': 0},
'max': {'heat.%CIRCUIT%:U_HDS_1_B1:first': 1000,
'heat.%CIRCUIT%:U_HDS_1_B2:first': 1000,
'heat.%CIRCUIT%:U_HDS_1_B1:last20mean': 10,
'heat.%CIRCUIT%:U_HDS_1_B2:last20mean': 10}})
pd.testing.assert_frame_equal(features_ref, features_act)
def test_read_quench_heaters_reference_features_ipq4(self):
# arrange
circuit_type = 'IPQ4'
# act
features_act = Reference.read_quench_heaters_reference_features(circuit_type)
# assert
features_ref = pd.DataFrame({'min': {'heat.%CIRCUIT%:U_HDS_1_B1:first': 810,
'heat.%CIRCUIT%:U_HDS_1_B2:first': 810,
'heat.%CIRCUIT%:U_HDS_2_B1:first': 810,
'heat.%CIRCUIT%:U_HDS_2_B2:first': 810,
'heat.%CIRCUIT%:U_HDS_1_B1:last20mean': 0,
'heat.%CIRCUIT%:U_HDS_1_B2:last20mean': 0,
'heat.%CIRCUIT%:U_HDS_2_B1:last20mean': 0,
'heat.%CIRCUIT%:U_HDS_2_B2:last20mean': 0},
'max': {'heat.%CIRCUIT%:U_HDS_1_B1:first': 1000,
'heat.%CIRCUIT%:U_HDS_1_B2:first': 1000,
'heat.%CIRCUIT%:U_HDS_2_B1:first': 1000,
'heat.%CIRCUIT%:U_HDS_2_B2:first': 1000,
'heat.%CIRCUIT%:U_HDS_1_B1:last20mean': 10,
'heat.%CIRCUIT%:U_HDS_1_B2:last20mean': 10,
'heat.%CIRCUIT%:U_HDS_2_B1:last20mean': 10,
'heat.%CIRCUIT%:U_HDS_2_B2:last20mean': 10}})
pd.testing.assert_frame_equal(features_ref, features_act)
def test_read_quench_heaters_reference_features_ipq8(self):
# arrange
circuit_type = 'IPQ8'
# act
features_act = Reference.read_quench_heaters_reference_features(circuit_type)
# assert
features_ref = pd.DataFrame({'min': {'heat.%CIRCUIT%:U_HDS_1_B1:first': 810,
'heat.%CIRCUIT%:U_HDS_1_B2:first': 810,
'heat.%CIRCUIT%:U_HDS_2_B1:first': 810,
'heat.%CIRCUIT%:U_HDS_2_B2:first': 810,
'heat.%CIRCUIT%:U_HDS_3_B1:first': 810,
'heat.%CIRCUIT%:U_HDS_3_B2:first': 810,
'heat.%CIRCUIT%:U_HDS_4_B1:first': 810,
'heat.%CIRCUIT%:U_HDS_4_B2:first': 810,
'heat.%CIRCUIT%:U_HDS_1_B1:last20mean': 0,
'heat.%CIRCUIT%:U_HDS_1_B2:last20mean': 0,
'heat.%CIRCUIT%:U_HDS_2_B1:last20mean': 0,
'heat.%CIRCUIT%:U_HDS_2_B2:last20mean': 0,
'heat.%CIRCUIT%:U_HDS_3_B1:last20mean': 0,
'heat.%CIRCUIT%:U_HDS_3_B2:last20mean': 0,
'heat.%CIRCUIT%:U_HDS_4_B1:last20mean': 0,
'heat.%CIRCUIT%:U_HDS_4_B2:last20mean': 0},
'max': {'heat.%CIRCUIT%:U_HDS_1_B1:first': 1000,
'heat.%CIRCUIT%:U_HDS_1_B2:first': 1000,
'heat.%CIRCUIT%:U_HDS_2_B1:first': 1000,
'heat.%CIRCUIT%:U_HDS_2_B2:first': 1000,
'heat.%CIRCUIT%:U_HDS_3_B1:first': 1000,
'heat.%CIRCUIT%:U_HDS_3_B2:first': 1000,
'heat.%CIRCUIT%:U_HDS_4_B1:first': 1000,
'heat.%CIRCUIT%:U_HDS_4_B2:first': 1000,
'heat.%CIRCUIT%:U_HDS_1_B1:last20mean': 10,
'heat.%CIRCUIT%:U_HDS_1_B2:last20mean': 10,
'heat.%CIRCUIT%:U_HDS_2_B1:last20mean': 10,
'heat.%CIRCUIT%:U_HDS_2_B2:last20mean': 10,
'heat.%CIRCUIT%:U_HDS_3_B1:last20mean': 10,
'heat.%CIRCUIT%:U_HDS_3_B2:last20mean': 10,
'heat.%CIRCUIT%:U_HDS_4_B1:last20mean': 10,
'heat.%CIRCUIT%:U_HDS_4_B2:last20mean': 10}})
pd.testing.assert_frame_equal(features_ref, features_act)
def test_read_quench_heaters_reference_differences_rb(self):
# arrange
circuit_type = 'RB'
# act
features_act = Reference.read_quench_heaters_reference_differences(circuit_type)
# assert
features_ref = pd.DataFrame({'diff': {'%CELL%:U_HDS_1:tau_charge': 0.003, '%CELL%:U_HDS_2:tau_charge': 0.003,
'%CELL%:U_HDS_3:tau_charge': 0.003, '%CELL%:U_HDS_4:tau_charge': 0.003,
'%CELL%:I_HDS_1:tau_charge': 0.003, '%CELL%:I_HDS_2:tau_charge': 0.003,
'%CELL%:I_HDS_3:tau_charge': 0.003, '%CELL%:I_HDS_4:tau_charge': 0.003,
'%CELL%:R_HDS_1:first20mean': 0.5, '%CELL%:R_HDS_2:first20mean': 0.5,
'%CELL%:R_HDS_3:first20mean': 0.5, '%CELL%:R_HDS_4:first20mean': 0.5}})
features_ref.sort_index(inplace=True)
features_act.sort_index(inplace=True)
pd.testing.assert_frame_equal(features_ref, features_act)
def test_read_quench_heaters_reference_differences_rq(self):
# arrange
circuit_type = 'RQ'
# act
features_act = Reference.read_quench_heaters_reference_differences(circuit_type)
# assert
features_ref = pd.DataFrame({'diff': {'%CELL%:U_HDS_1:tau_charge': 0.003, '%CELL%:U_HDS_2:tau_charge': 0.003}})
features_ref.sort_index(inplace=True)
features_act.sort_index(inplace=True)
pd.testing.assert_frame_equal(features_ref, features_act)
def test_read_quench_heaters_reference_differences_it(self):
# arrange
circuit_type = 'IT'
# act
features_act = Reference.read_quench_heaters_reference_differences(circuit_type)
# assert
features_ref = pd.DataFrame({'diff': {'heat.%CIRCUIT%:U_HDS_1_Q1:tau_charge': 0.005,
'heat.%CIRCUIT%:U_HDS_2_Q1:tau_charge': 0.005,
'heat.%CIRCUIT%:U_HDS_1_Q2:tau_charge': 0.005,
'heat.%CIRCUIT%:U_HDS_2_Q2:tau_charge': 0.005,
'heat.%CIRCUIT%:U_HDS_3_Q2:tau_charge': 0.005,
'heat.%CIRCUIT%:U_HDS_4_Q2:tau_charge': 0.005,
'heat.%CIRCUIT%:U_HDS_1_Q3:tau_charge': 0.005,
'heat.%CIRCUIT%:U_HDS_2_Q3:tau_charge': 0.005}})
features_ref.sort_index(inplace=True)
features_act.sort_index(inplace=True)
pd.testing.assert_frame_equal(features_ref, features_act)
def test_read_quench_heaters_reference_differences_ipd2(self):
# arrange
circuit_type = 'IPD2'
# act
features_act = Reference.read_quench_heaters_reference_differences(circuit_type)
# assert
features_ref = pd.DataFrame({'diff': {'heat.%CIRCUIT%:U_HDS_1:tau_charge': 0.005,
'heat.%CIRCUIT%:U_HDS_2:tau_charge': 0.005}})
features_ref.sort_index(inplace=True)
features_act.sort_index(inplace=True)
pd.testing.assert_frame_equal(features_ref, features_act)
def test_read_quench_heaters_reference_differences_ipd2_b1b2(self):
# arrange
circuit_type = 'IPD2_B1B2'
# act
features_act = Reference.read_quench_heaters_reference_differences(circuit_type)
# assert
features_ref = pd.DataFrame({'diff': {'heat.%CIRCUIT%:U_HDS_1_B1:tau_charge': 0.005,
'heat.%CIRCUIT%:U_HDS_1_B2:tau_charge': 0.005}})
features_ref.sort_index(inplace=True)
features_act.sort_index(inplace=True)
pd.testing.assert_frame_equal(features_ref, features_act)
def test_read_quench_heaters_reference_differences_ipq2(self):
# arrange
circuit_type = 'IPQ2'
# act
features_act = Reference.read_quench_heaters_reference_differences(circuit_type)
# assert
features_ref = pd.DataFrame({'diff': {'heat.%CIRCUIT%:U_HDS_1_B1:tau_charge': 0.005,
'heat.%CIRCUIT%:U_HDS_1_B2:tau_charge': 0.005}})
features_ref.sort_index(inplace=True)
features_act.sort_index(inplace=True)
pd.testing.assert_frame_equal(features_ref, features_act)
def test_read_quench_heaters_reference_differences_ipq4(self):
# arrange
circuit_type = 'IPQ4'
# act
features_act = Reference.read_quench_heaters_reference_differences(circuit_type)
# assert
features_ref = pd.DataFrame({'diff': {'heat.%CIRCUIT%:U_HDS_1_B1:tau_charge': 0.005,
'heat.%CIRCUIT%:U_HDS_1_B2:tau_charge': 0.005,
'heat.%CIRCUIT%:U_HDS_2_B1:tau_charge': 0.005,
'heat.%CIRCUIT%:U_HDS_2_B2:tau_charge': 0.005}})
features_ref.sort_index(inplace=True)
features_act.sort_index(inplace=True)
pd.testing.assert_frame_equal(features_ref, features_act)
def test_read_quench_heaters_reference_differences_ipq8(self):
# arrange
circuit_type = 'IPQ8'
# act
features_act = Reference.read_quench_heaters_reference_differences(circuit_type)
# assert
features_ref = pd.DataFrame({'diff': {'heat.%CIRCUIT%:U_HDS_1_B1:tau_charge': 0.005,
'heat.%CIRCUIT%:U_HDS_1_B2:tau_charge': 0.005,
'heat.%CIRCUIT%:U_HDS_2_B1:tau_charge': 0.005,
'heat.%CIRCUIT%:U_HDS_2_B2:tau_charge': 0.005,
'heat.%CIRCUIT%:U_HDS_3_B1:tau_charge': 0.005,
'heat.%CIRCUIT%:U_HDS_3_B2:tau_charge': 0.005,
'heat.%CIRCUIT%:U_HDS_4_B1:tau_charge': 0.005,
'heat.%CIRCUIT%:U_HDS_4_B2:tau_charge': 0.005}})
features_ref.sort_index(inplace=True)
features_act.sort_index(inplace=True)
pd.testing.assert_frame_equal(features_ref, features_act)
self.assertEqual(period_exp, period_act)
# Support to several reference QH discharges to account for magnet replacement over time
def test_get_quench_heater_reference_discharge_b24l5(self):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment