@@ -450,6 +450,12 @@ def test_3to4_deprecated_magnetics(dd4factory):
450450 mag .bpol_probe [1 ].name = "name2"
451451 mag .bpol_probe [1 ].voltage .data = [0.1 , 0.2 , 0.3 ]
452452
453+ mag .method .resize (2 )
454+ for i , method in enumerate (mag .method ):
455+ method .name = f"name{ i } "
456+ method .ip .data = [i , 1.0 , 2.0 ]
457+ method .ip .time = [i + 1 , 2.0 , 3.0 ]
458+
453459 mag4 = convert_ids (mag , None , factory = dd4factory )
454460 assert len (mag4 .b_field_pol_probe ) == 2
455461 assert mag4 .b_field_pol_probe [0 ].name == "identifier1"
@@ -460,14 +466,26 @@ def test_3to4_deprecated_magnetics(dd4factory):
460466 assert mag4 .b_field_pol_probe [1 ].description == "name2"
461467 assert array_equal (mag4 .b_field_pol_probe [1 ].voltage .data , [0.1 , 0.2 , 0.3 ])
462468
469+ assert len (mag4 .ip ) == 2
470+ assert mag4 .ip [0 ].method_name == "name0"
471+ assert array_equal (mag4 .ip [0 ].data , [0.0 , 1.0 , 2.0 ])
472+ assert array_equal (mag4 .ip [0 ].time , [1.0 , 2.0 , 3.0 ])
473+ assert mag4 .ip [1 ].method_name == "name1"
474+ assert array_equal (mag4 .ip [1 ].data , [1.0 , 1.0 , 2.0 ])
475+ assert array_equal (mag4 .ip [1 ].time , [2.0 , 2.0 , 3.0 ])
476+
463477 # If both the deprecated and the "correct" quantity exist, we expect only the
464478 # correct one to be converted to DD4:
465479 mag .b_field_pol_probe .resize (1 )
466480 mag .b_field_pol_probe [0 ].name = "test"
481+ mag .ip .resize (1 )
482+ mag .ip [0 ].method_name = "ip"
467483
468484 mag4 = convert_ids (mag , None , factory = dd4factory )
469485 assert len (mag4 .b_field_pol_probe ) == 1
470486 assert mag4 .b_field_pol_probe [0 ].name == "test"
487+ assert len (mag4 .ip ) == 1
488+ assert mag4 .ip [0 ].method_name == "ip"
471489
472490
473491def test_3to4_pulse_schedule ():
0 commit comments