Skip to content

Commit f2054db

Browse files
bhazeltonmkolopanis
authored andcommitted
Add test coverage
1 parent da74973 commit f2054db

File tree

1 file changed

+24
-5
lines changed

1 file changed

+24
-5
lines changed

tests/uvbeam/test_feko_beam.py

Lines changed: 24 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)