@@ -101,15 +101,34 @@ def test_read_feko(btype):
101101
102102
103103@pytest .mark .parametrize (
104- ("feedpol " , "msg" ),
104+ ("kwargs " , "msg" ),
105105 [
106- (np .array ([["x" ]]), "feed_pol can not be a multi-dimensional array" ),
107- (["x" , "y" ], "feed_pol must have exactly one element" ),
106+ (
107+ {"feed_pol" : np .array ([["x" ]])},
108+ "feed_pol can not be a multi-dimensional array" ,
109+ ),
110+ ({"feed_pol" : ["x" , "y" ]}, "feed_pol must have exactly one element" ),
111+ (
112+ {"filename" : [filename_x , filename_y ]},
113+ "If multiple FEKO files are passed, the feed_pol must be a list "
114+ "or array of the same length giving the feed_pol for each file." ,
115+ ),
116+ (
117+ {
118+ "filename" : [filename_x , filename_y ],
119+ "feed_pol" : ["x" , "y" ],
120+ "feed_angle" : np .pi / 2 ,
121+ },
122+ "If multiple FEKO files are passed, the feed_angle must be a list or "
123+ "array of the same length giving the feed_angle for each file." ,
124+ ),
108125 ],
109126)
110- def test_read_feko_feedpol_errors (feedpol , msg ):
127+ def test_read_feko_input_errors (kwargs , msg ):
128+ init_kwargs = {"filename" : filename_x , "beam_type" : "power" , "feed_pol" : "x" }
129+ init_kwargs .update (kwargs )
111130 with pytest .raises (ValueError , match = msg ):
112- UVBeam .from_file (feko_filename_x , beam_type = "power" , feed_pol = feedpol )
131+ UVBeam .from_file (** init_kwargs )
113132
114133
115134@pytest .mark .parametrize (
0 commit comments