Fixed some errors and removed tests moved to pandablocks-client

src/pandablocks_ioc/_tables.py

@@ -312,16 +312,12 @@ def __init__(
             full_name = EpicsName(full_name)
             description = trim_description(field_details.description, full_name)
 
-            waveform_val = self._construct_waveform_val(
-                field_data, field_name, field_details
-            )
-
             field_record: RecordWrapper = builder.WaveformOut(
                 full_name,
                 DESC=description,
                 validate=self.validate_waveform,
                 on_update_name=self.update_waveform,
-                initial_value=waveform_val,
+                initial_value=field_data[field_name],
                 length=field_info.max_length,
             )
 
@@ -483,23 +479,6 @@ def __init__(
             SignalRW(index_record_name, index_record_name, TextWrite()),
         )
 
-    def _construct_waveform_val(
-        self,
-        field_data: Dict[str, UnpackedArray],
-        field_name: str,
-        field_details: TableFieldDetails,
-    ):
-        """Convert the values into the right form. For enums this means converting
-        the numeric values PandA sends us into the string representation. For all other
-        types the numeric representation is used."""
-
-        if field_details.labels and not all(
-            [isinstance(x, str) for x in field_data[field_name]]
-        ):
-            return [field_details.labels[x] for x in field_data[field_name]]
-
-        return field_data[field_name]
-
     def validate_waveform(self, record: RecordWrapper, new_val) -> bool:
         """Controls whether updates to the waveform records are processed, based on the
         value of the MODE record.
@@ -565,7 +544,14 @@ async def update_mode(self, new_val: int):
             try:
                 # Send all EPICS data to PandA
                 logging.info(f"Sending table data for {self.table_name} to PandA")
-                packed_data = table_to_words(self.all_values_dict, self.field_info)
+
+                table = {}
+                for x in self.table_fields_records:
+                    record_info = self.table_fields_records[x].record_info
+                    if record_info:
+                        table[x] = record_info.record.get()
+
+                packed_data = table_to_words(table, self.field_info)
 
                 panda_field_name = epics_to_panda_name(self.table_name)
                 await self.client.send(Put(panda_field_name, packed_data))
@@ -638,11 +624,11 @@ def update_table(self, new_values: List[str]) -> None:
 
             for field_name, field_record in self.table_fields_records.items():
                 assert field_record.record_info
-                waveform_val = self._construct_waveform_val(
-                    field_data, field_name, field_record.field
-                )
+
                 # Must skip processing as the validate method would reject the update
-                field_record.record_info.record.set(waveform_val, process=False)
+                field_record.record_info.record.set(
+                    field_data[field_name], process=False
+                )
                 self._update_scalar(field_record.record_info.record.name)
 
             # All items in field_data have the same length, so just use 0th.

tests/test_tables.py

@@ -312,87 +312,6 @@ async def test_create_softioc_update_table_scalars_change(
         repeats_monitor.close()
 
 
-def test_table_packing_unpack(
-    table_field_info: TableFieldInfo,
-    table_fields_records: Dict[str, TableFieldRecordContainer],
-    table_data_1: List[str],
-    table_unpacked_data: typing.OrderedDict[EpicsName, ndarray],
-):
-    """Test table unpacking works as expected"""
-    assert table_field_info.row_words
-    unpacked = TablePacking.unpack(
-        table_field_info.row_words, table_fields_records, table_data_1
-    )
-
-    actual: Union[UnpackedArray, List[str]]
-    for field_name, actual in unpacked.items():
-        expected = table_unpacked_data[EpicsName(field_name)]
-        if expected.dtype.char in ("S", "U"):
-            # Convert numeric array back to strings
-            actual = [table_fields_records[field_name].field.labels[x] for x in actual]
-        numpy.testing.assert_array_equal(actual, expected)
-
-
-def test_table_packing_pack(
-    table_field_info: TableFieldInfo,
-    table_fields_records: Dict[str, TableFieldRecordContainer],
-    table_data_1: List[str],
-):
-    """Test table unpacking works as expected"""
-    assert table_field_info.row_words
-    unpacked = TablePacking.pack(table_field_info.row_words, table_fields_records)
-
-    for actual, expected in zip(unpacked, table_data_1):
-        assert actual == expected
-
-
-def test_table_packing_pack_length_mismatched(
-    table_field_info: TableFieldInfo,
-    table_fields_records: Dict[str, TableFieldRecordContainer],
-):
-    """Test that mismatching lengths on waveform inputs causes an exception"""
-    assert table_field_info.row_words
-
-    # Adjust one of the record lengths so it mismatches
-    record_info = table_fields_records[EpicsName("OUTC1")].record_info
-    assert record_info
-    record_info.record.get = MagicMock(return_value=array([1, 2, 3, 4, 5, 6, 7, 8]))
-
-    with pytest.raises(AssertionError):
-        TablePacking.pack(table_field_info.row_words, table_fields_records)
-
-
-def test_table_packing_roundtrip(
-    table_field_info: TableFieldInfo,
-    table_fields: Dict[str, TableFieldDetails],
-    table_fields_records: Dict[str, TableFieldRecordContainer],
-    table_data_1: List[str],
-):
-    """Test that calling unpack -> pack yields the same data"""
-    assert table_field_info.row_words
-    unpacked = TablePacking.unpack(
-        table_field_info.row_words, table_fields_records, table_data_1
-    )
-
-    # Put these values into Mocks for the Records
-    data: Dict[str, TableFieldRecordContainer] = {}
-    for field_name, field_info in table_fields.items():
-        return_value: Union[UnpackedArray, List[str]] = unpacked[field_name]
-        if field_info.labels:
-            # Convert to string representation
-            return_value = [field_info.labels[x] for x in return_value]
-
-        mocked_record = MagicMock()
-        mocked_record.get = MagicMock(return_value=return_value)
-        record_info = RecordInfo(lambda x: None)
-        record_info.add_record(mocked_record)
-        data[field_name] = TableFieldRecordContainer(field_info, record_info)
-
-    packed = TablePacking.pack(table_field_info.row_words, data)
-
-    assert packed == table_data_1
-
-
 def test_table_updater_validate_mode_view(table_updater: TableUpdater):
     """Test the validate method when mode is View"""
 
@@ -508,9 +427,6 @@ async def test_table_updater_update_mode_submit_exception(
         called_args = record_info.record.set.call_args
 
         expected = called_args[0][0]
-        labels = table_updater.table_fields_records[field_name].field.labels
-        if labels:
-            expected = [labels[x] for x in expected]
 
         numpy.testing.assert_array_equal(data, expected)
 
@@ -569,10 +485,6 @@ async def test_table_updater_update_mode_discard(
 
         expected = called_args[0][0]
 
-        labels = table_updater.table_fields_records[field_name].field.labels
-        if labels:
-            expected = [labels[x] for x in expected]
-
         numpy.testing.assert_array_equal(data, expected)
 
     table_updater.mode_record_info.record.set.assert_called_once_with(