Add functions to GenericLookupTable

Author: jacob720

src/daq_config_server/converters/models.py

@@ -45,3 +45,24 @@ def check_row_length_matches_n_columns(self):
                     f"of columns: {self.column_names}"
                 )
         return self
+
+    def get_value(
+        self,
+        column_name: str,
+        value: int | float,
+        target_column_name: str,
+        value_must_exist: bool = True,
+    ) -> int | float:
+        column_index = self.column_names.index(column_name)
+        column = [row[column_index] for row in self.rows]
+        target_column_index = self.column_names.index(target_column_name)
+
+        closest_value = (
+            min(column, key=lambda x: abs(x - value)) if not value_must_exist else value
+        )
+        target_row = self.rows[column.index(closest_value)]
+
+        return target_row[target_column_index]
+
+    def columns(self):
+        return [[row[i] for row in self.rows] for i in range(len(self.column_names))]

tests/unit_tests/test_converters.py

@@ -190,7 +190,7 @@ def test_bad_beamline_parameters_with_non_keyword_string_value_causes_error():
         beamline_parameters_to_dict(contents)
 
 
-def test_beam_line_parameters_with_repeated_key_causes_error():
+def test_beamline_parameters_with_repeated_key_causes_error():
     input = "thing = 1\nthing = 2"
     with pytest.raises(ValueError, match="Repeated key in parameters: thing"):
         beamline_parameters_to_dict(input)
@@ -247,6 +247,33 @@ def test_detector_xy_lut_gives_expected_results():
     assert result == expected
 
 
+def test_generic_lut_model_get_value_function():
+    my_lut = GenericLookupTable(
+        column_names=["detector_distances_mm", "beam_centre_x_mm", "beam_centre_y_mm"],
+        rows=[[150, 152.2, 166.26], [800, 152.08, 160.96]],
+    )
+    assert my_lut.get_value("detector_distances_mm", 150, "beam_centre_x_mm") == 152.2
+    assert my_lut.get_value("beam_centre_y_mm", 160.96, "detector_distances_mm") == 800
+    with pytest.raises(ValueError):
+        # value doesn't exist
+        my_lut.get_value("beam_centre_y_mm", 160.97, "detector_distances_mm")
+    assert (
+        my_lut.get_value(
+            "beam_centre_x_mm", 153, "beam_centre_y_mm", value_must_exist=False
+        )
+        == 166.26  # get closest value
+    )
+
+
+def test_generic_lut_model_columns_function():
+    my_lut = GenericLookupTable(
+        column_names=["detector_distances_mm", "beam_centre_x_mm", "beam_centre_y_mm"],
+        rows=[[150, 152.2, 166.26], [800, 152.08, 160.96]],
+    )
+    expected_columns = [[150, 800], [152.2, 152.08], [166.26, 160.96]]
+    assert my_lut.columns() == expected_columns
+
+
 def test_beamline_pitch_lut_gives_expected_result():
     input = (
         "#       Bragg	pitch\n"