Skip to content

Commit c42764f

Browse files
author
Michael Johansen
committed
Add a few missing test cases. Rename tests for consistency.
Signed-off-by: Michael Johansen <[email protected]>
1 parent 799356a commit c42764f

File tree

1 file changed

+22
-19
lines changed

1 file changed

+22
-19
lines changed

tests/unit/test_convert.py

Lines changed: 22 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,6 @@
1414
import nipanel._convert
1515
import tests.types
1616

17-
1817
_AnyWrappersPb2: TypeAlias = Union[
1918
wrappers_pb2.BoolValue,
2019
wrappers_pb2.BytesValue,
@@ -130,6 +129,8 @@
130129
ComplexWaveform(0, ComplexInt32DType),
131130
"nitypes.waveform.ComplexWaveform[[('real', '<i2'), ('imag', '<i2')]]",
132131
),
132+
(DigitalWaveform(10, 2, np.bool, False), "nitypes.waveform.DigitalWaveform"),
133+
(Spectrum(10, np.float64), "nitypes.waveform.Spectrum"),
133134
],
134135
)
135136
def test___various_python_objects___get_best_matching_type___returns_correct_type_string(
@@ -251,7 +252,7 @@ def test___durationpb2_timestamp___from_any___valid_python_value() -> None:
251252
(array_pb2.StringArray, ["", "", ""], ["a", "b", "c"]),
252253
],
253254
)
254-
def test___python_panel_collection___to_any___valid_paneltype_value(
255+
def test___python_collection___to_any___valid_array_proto(
255256
proto_type: type[_AnyPanelPbTypes], default_value: Any, expected_value: Any
256257
) -> None:
257258
result = nipanel._convert.to_any(expected_value)
@@ -262,7 +263,7 @@ def test___python_panel_collection___to_any___valid_paneltype_value(
262263
assert unpack_dest.values == expected_value
263264

264265

265-
def test___python_scalar_object___to_any___valid_scalar_data_value() -> None:
266+
def test___python_scalar_object___to_any___valid_scalar_proto() -> None:
266267
scalar_obj = Scalar(1.0, "amps")
267268
result = nipanel._convert.to_any(scalar_obj)
268269
unpack_dest = scalar_pb2.Scalar()
@@ -273,7 +274,7 @@ def test___python_scalar_object___to_any___valid_scalar_data_value() -> None:
273274
assert unpack_dest.attributes["NI_UnitDescription"].string_value == "amps"
274275

275276

276-
def test___python_float64_analog_waveform___to_any___valid_double_analog_waveform() -> None:
277+
def test___python_float64_analog_waveform___to_any___valid_double_analog_waveform_proto() -> None:
277278
wfm_obj = AnalogWaveform(3, np.float64)
278279
result = nipanel._convert.to_any(wfm_obj)
279280
unpack_dest = waveform_pb2.DoubleAnalogWaveform()
@@ -283,7 +284,7 @@ def test___python_float64_analog_waveform___to_any___valid_double_analog_wavefor
283284
assert list(unpack_dest.y_data) == [0.0, 0.0, 0.0]
284285

285286

286-
def test___python_int16_analog_waveform___to_any___valid_i16_analog_waveform() -> None:
287+
def test___python_int16_analog_waveform___to_any___valid_i16_analog_waveform_proto() -> None:
287288
wfm_obj = AnalogWaveform(3, np.int16)
288289
result = nipanel._convert.to_any(wfm_obj)
289290
unpack_dest = waveform_pb2.I16AnalogWaveform()
@@ -293,7 +294,7 @@ def test___python_int16_analog_waveform___to_any___valid_i16_analog_waveform() -
293294
assert list(unpack_dest.y_data) == [0, 0, 0]
294295

295296

296-
def test___python_float64_complex_waveform___to_any___valid_double_complex_waveform() -> None:
297+
def test___python_float64_complex_waveform___to_any___valid_double_complex_waveform_proto() -> None:
297298
wfm_obj = ComplexWaveform(2, np.complex128)
298299
result = nipanel._convert.to_any(wfm_obj)
299300
unpack_dest = waveform_pb2.DoubleComplexWaveform()
@@ -303,7 +304,7 @@ def test___python_float64_complex_waveform___to_any___valid_double_complex_wavef
303304
assert list(unpack_dest.y_data) == [0.0, 0.0, 0.0, 0.0]
304305

305306

306-
def test___python_int16_complex_waveform___to_any___valid_i16_complex_waveform() -> None:
307+
def test___python_int16_complex_waveform___to_any___valid_i16_complex_waveform_proto() -> None:
307308
wfm_obj = ComplexWaveform(2, ComplexInt32DType)
308309
result = nipanel._convert.to_any(wfm_obj)
309310
unpack_dest = waveform_pb2.I16ComplexWaveform()
@@ -313,7 +314,7 @@ def test___python_int16_complex_waveform___to_any___valid_i16_complex_waveform()
313314
assert list(unpack_dest.y_data) == [0, 0, 0, 0]
314315

315316

316-
def test___bool_digital_waveform___to_any___valid_digital_waveform_protobuf() -> None:
317+
def test___python_bool_digital_waveform___to_any___valid_digital_waveform_proto() -> None:
317318
data = np.array([[0, 1, 0], [1, 0, 1]], dtype=np.bool)
318319
wfm_obj = DigitalWaveform.from_lines(data, signal_count=3)
319320

@@ -326,7 +327,7 @@ def test___bool_digital_waveform___to_any___valid_digital_waveform_protobuf() ->
326327
assert unpack_dest.signal_count == 3
327328

328329

329-
def test___uint8_digital_waveform___to_any___valid_digital_waveform_protobuf() -> None:
330+
def test___python_uint8_digital_waveform___to_any___valid_digital_waveform_proto() -> None:
330331
data = np.array([[0, 1, 3], [7, 5, 1]], dtype=np.uint8)
331332
wfm_obj = DigitalWaveform.from_lines(data, signal_count=3)
332333

@@ -339,7 +340,7 @@ def test___uint8_digital_waveform___to_any___valid_digital_waveform_protobuf() -
339340
assert unpack_dest.signal_count == 3
340341

341342

342-
def test___float64_spectrum___to_any___valid_double_spectrum_protobuf() -> None:
343+
def test___python_float64_spectrum___to_any___valid_double_spectrum_proto() -> None:
343344
spectrum = Spectrum.from_array_1d(np.array([1.0, 2.0, 3.0]))
344345
spectrum.start_frequency = 100.0
345346
spectrum.frequency_increment = 10.0
@@ -424,7 +425,7 @@ def test___python_set_of_collection_of_float___to_any___valid_double2darray(
424425
(array_pb2.StringArray, ["a", "b", "c"]),
425426
],
426427
)
427-
def test___paneltype_value___from_any___valid_python_value(
428+
def test___array_proto___from_any___valid_python_collection(
428429
proto_type: type[_AnyPanelPbTypes], expected_value: Any
429430
) -> None:
430431
pb_value = proto_type(values=expected_value)
@@ -436,7 +437,7 @@ def test___paneltype_value___from_any___valid_python_value(
436437
assert result == expected_value
437438

438439

439-
def test___scalar_data___from_any___valid_python_scalar_object() -> None:
440+
def test___scalar_proto___from_any___valid_python_scalar() -> None:
440441
attrs = {"NI_UnitDescription": attribute_value_pb2.AttributeValue(string_value="amps")}
441442
pb_value = scalar_pb2.Scalar(attributes=attrs, double_value=1.0)
442443
packed_any = _pack_into_any(pb_value)
@@ -448,7 +449,7 @@ def test___scalar_data___from_any___valid_python_scalar_object() -> None:
448449
assert result.units == "amps"
449450

450451

451-
def test___double_analog_waveform___from_any___valid_python_float64_analog_waveform() -> None:
452+
def test___double_analog_waveform_proto___from_any___valid_python_float64_analog_waveform() -> None:
452453
pb_value = waveform_pb2.DoubleAnalogWaveform(y_data=[0.0, 0.0, 0.0])
453454
packed_any = _pack_into_any(pb_value)
454455

@@ -459,7 +460,7 @@ def test___double_analog_waveform___from_any___valid_python_float64_analog_wavef
459460
assert result.dtype == np.float64
460461

461462

462-
def test___i16_analog_waveform___from_any___valid_python_int16_analog_waveform() -> None:
463+
def test___i16_analog_waveform_proto___from_any___valid_python_int16_analog_waveform() -> None:
463464
pb_value = waveform_pb2.I16AnalogWaveform(y_data=[0, 0, 0])
464465
packed_any = _pack_into_any(pb_value)
465466

@@ -470,7 +471,9 @@ def test___i16_analog_waveform___from_any___valid_python_int16_analog_waveform()
470471
assert result.dtype == np.int16
471472

472473

473-
def test___double_complex_waveform___from_any___valid_python_float64_complex_waveform() -> None:
474+
def test___double_complex_waveform_proto___from_any___valid_python_float64_complex_waveform() -> (
475+
None
476+
):
474477
pb_value = waveform_pb2.DoubleComplexWaveform(y_data=[0.0, 0.0, 0.0, 0.0])
475478
packed_any = _pack_into_any(pb_value)
476479

@@ -481,7 +484,7 @@ def test___double_complex_waveform___from_any___valid_python_float64_complex_wav
481484
assert result.dtype == np.complex128
482485

483486

484-
def test___i16_complex_waveform___from_any___valid_python_int16_complex_waveform() -> None:
487+
def test___i16_complex_waveform_proto___from_any___valid_python_int16_complex_waveform() -> None:
485488
pb_value = waveform_pb2.I16ComplexWaveform(y_data=[0, 0, 0, 0])
486489
packed_any = _pack_into_any(pb_value)
487490

@@ -492,7 +495,7 @@ def test___i16_complex_waveform___from_any___valid_python_int16_complex_waveform
492495
assert result.dtype == ComplexInt32DType
493496

494497

495-
def test___digital_waveform_proto___from_any___valid_bool_digital_waveform_object() -> None:
498+
def test___digital_waveform_proto___from_any___valid_python_bool_digital_waveform() -> None:
496499
data = np.array([[0, 1, 0], [1, 0, 1]], dtype=np.bool)
497500
pb_value = waveform_pb2.DigitalWaveform(y_data=data.tobytes(), signal_count=3)
498501
packed_any = _pack_into_any(pb_value)
@@ -504,7 +507,7 @@ def test___digital_waveform_proto___from_any___valid_bool_digital_waveform_objec
504507
assert result.signal_count == 3
505508

506509

507-
def test___digital_waveform_proto___from_any___valid_uint8_digital_waveform_object() -> None:
510+
def test___digital_waveform_proto___from_any___valid_python_uint8_digital_waveform() -> None:
508511
data = np.array([[0, 1, 0], [1, 0, 1]], dtype=np.uint8)
509512
pb_value = waveform_pb2.DigitalWaveform(y_data=data.tobytes(), signal_count=3)
510513
packed_any = _pack_into_any(pb_value)
@@ -516,7 +519,7 @@ def test___digital_waveform_proto___from_any___valid_uint8_digital_waveform_obje
516519
assert result.signal_count == 3
517520

518521

519-
def test___double_spectrum_proto___from_any___valid_spectrum_object() -> None:
522+
def test___double_spectrum_proto___from_any___valid_python_spectrum() -> None:
520523
pb_value = waveform_pb2.DoubleSpectrum(
521524
data=[1.0, 2.0, 3.0],
522525
start_frequency=100.0,

0 commit comments

Comments
 (0)