@@ -101,15 +101,34 @@ def test_read_feko(btype):
101
101
102
102
103
103
@pytest .mark .parametrize (
104
- ("feedpol " , "msg" ),
104
+ ("kwargs " , "msg" ),
105
105
[
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
+ ),
108
125
],
109
126
)
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 )
111
130
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 )
113
132
114
133
115
134
@pytest .mark .parametrize (
0 commit comments