@@ -363,16 +363,15 @@ def make_antenna_sim():
363363 remove_dc_component = False , # Include DC component for more accuracy at low frequencies
364364 )
365365
366- sim_unit = list (modeler .sim_dict .values ())[0 ]
367-
368- return sim_unit
366+ return modeler
369367
370368
371369def test_rectangular_array_calculator_array_make_antenna_array ():
372370 """Test automatic antenna array creation."""
373371 freq0 = 10e9
374372 wavelength0 = td .C_0 / 10e9
375- sim_unit = make_antenna_sim ()
373+ modeler = make_antenna_sim ()
374+ sim_unit = list (modeler .sim_dict .values ())[0 ]
376375 array_calculator = mw .RectangularAntennaArrayCalculator (
377376 array_size = (1 , 2 , 3 ),
378377 spacings = (0.5 * wavelength0 , 0.6 * wavelength0 , 0.4 * wavelength0 ),
@@ -437,8 +436,9 @@ def test_rectangular_array_calculator_array_make_antenna_array():
437436 assert len (sim_array .sources ) == 6
438437
439438 # check that override_structures are duplicated
440- assert len (sim_unit .grid_spec .override_structures ) == 2
441- assert len (sim_array .grid_spec .override_structures ) == 7
439+ # assert len(sim_unit.grid_spec.override_structures) == 2
440+ # assert len(sim_array.grid_spec.override_structures) == 7
441+ assert sim_unit .grid .boundaries == modeler .base_sim .grid .boundaries
442442
443443 # check that phase shifts are applied correctly
444444 phases_expected = array_calculator ._antenna_phases
@@ -674,7 +674,8 @@ def test_rectangular_array_calculator_simulation_data_from_array_factor():
674674 phase_shifts = (np .pi / 3 , np .pi / 4 , np .pi / 5 ),
675675 )
676676
677- sim_unit = make_antenna_sim ()
677+ modeler = make_antenna_sim ()
678+ sim_unit = list (modeler .sim_dict .values ())[0 ]
678679
679680 monitor = sim_unit .monitors [0 ]
680681 monitor_directivity = sim_unit .monitors [2 ]
0 commit comments