Skip to content

Commit

Permalink
test: fix test calls of BeamInterface
Browse files Browse the repository at this point in the history
  • Loading branch information
steven-murray committed Nov 22, 2024
1 parent 9c36b9b commit 3ac61dd
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions tests/test_beams.py
Original file line number Diff line number Diff line change
Expand Up @@ -143,10 +143,10 @@ def setup_class(self):
self.gpuinterp = GPUBeamInterpolator

@pytest.mark.skipif(not HAVE_GPU, reason="GPU is not available")
def test_exceptions(self, efield_single_freq: UVBeam):
def test_exceptions(self, efield_single_freq: BeamInterface):
"""Test that proper exceptions are raised when bad params are passed."""
beam = efield_single_freq.copy()
beam.to_healpix()
beam = efield_single_freq
beam = beam.clone(beam=beam.beam.to_healpix(inplace=False))
bm = self.gpuinterp(
beam_list=[beam],
beam_idx=np.zeros(1, dtype=int),
Expand Down Expand Up @@ -221,8 +221,8 @@ def test_gpu_beam_interp_against_cpu(efield_single_freq):
)

np.testing.assert_allclose(
cpu_bmfunc.beam_list[0].data_array,
gpu_bmfunc.beam_list[0].data_array,
cpu_bmfunc.beam_list[0].beam.data_array,
gpu_bmfunc.beam_list[0].beam.data_array,
atol=1e-8,
)
cpu_bmfunc.setup()
Expand Down

0 comments on commit 3ac61dd

Please sign in to comment.