diff --git a/tests/test_wollam.py b/tests/test_wollam.py index a7b75443..e7c3a6d0 100644 --- a/tests/test_wollam.py +++ b/tests/test_wollam.py @@ -1,28 +1,43 @@ """Tests for reading woollam""" +import numpy as np import pytest +import xarray as xr from fixtures import datadir # pylint: disable=unused-import + import elli # pylint: disable=redefined-outer-name def test_reading_of_psi_delta_woollam(datadir): - """Psi/delta Spectraray file is read w/o errors""" + """Psi/delta Wollam file is read w/o errors""" data_wvase = elli.read_woollam_psi_delta(datadir / "wvase_example.dat") data_cease = elli.read_woollam_psi_delta(datadir / "complete_ease_example.dat") - assert data_wvase.shape == (542, 2) - assert data_cease.shape == (3263, 2) + assert len(data_wvase) == 2 + assert len(data_cease) == 2 + + assert len(data_wvase.Wavelength) == 181 + assert len(data_cease.Wavelength) == 1088 + + np.testing.assert_array_equal(data_wvase["Angle of Incidence"], [65, 70, 75]) + np.testing.assert_array_equal(data_cease["Angle of Incidence"], [50, 60, 70]) # pylint: disable=redefined-outer-name def test_reading_and_conv_to_woollam(datadir): - """Rho values are read from Psi / Delta file""" + """Rho values are calculated from Psi / Delta file""" data_wvase = elli.read_woollam_rho(datadir / "wvase_example.dat") data_cease = elli.read_woollam_rho(datadir / "complete_ease_example.dat") - assert data_wvase.shape == (542,) - assert data_cease.shape == (3263,) + assert isinstance(data_wvase, xr.DataArray) + assert isinstance(data_cease, xr.DataArray) + + assert len(data_wvase.Wavelength) == 181 + assert len(data_cease.Wavelength) == 1088 + + np.testing.assert_array_equal(data_wvase["Angle of Incidence"], [65, 70, 75]) + np.testing.assert_array_equal(data_cease["Angle of Incidence"], [50, 60, 70]) # pylint: disable=redefined-outer-name