# import json
import numpy
import h5py
import pytest
from est.core.types import Spectrum, XASObject
from est.core.types.dimensions import transform_to_standard
from est.tests.data import example_spectra
from est.core.io import read_from_url
from silx.io.url import DataUrl
from est.units import ur
import silx.io.utils
[docs]
def test_from_dat(spectrum_cu_from_pymca):
"""check that we can create a Spectrum from a pymca .dat file"""
assert spectrum_cu_from_pymca.energy is not None
assert spectrum_cu_from_pymca.mu is not None
[docs]
def test_from_numpy_array():
"""check that we can create a Spectrum from numpy arrays"""
energy = numpy.arange(10, 20)
mu = numpy.arange(10)
spectrum = Spectrum(energy=energy, mu=mu)
numpy.testing.assert_array_equal(spectrum.energy, (energy * ur.eV).m)
numpy.testing.assert_array_equal(spectrum.mu, mu)
mu_2 = numpy.arange(30, 40)
spectrum.mu = mu_2
numpy.testing.assert_array_equal(spectrum.mu, mu_2)
[docs]
def test_create_from_single_spectrum(spectrum_cu_from_pymca):
"""check that we can create a XASObject from a pymca .dat file"""
configuration = {
"FT": {"KWeight": 1},
"EXAFS": {"EXAFSNormalized": numpy.array([1, 2, 3])},
}
obj = XASObject(
spectra=(spectrum_cu_from_pymca,),
energy=spectrum_cu_from_pymca.energy,
configuration=configuration,
dim1=1,
dim2=1,
)
assert obj.n_spectrum == 1
ddict = obj.to_dict()
obj2 = XASObject.from_dict(ddict)
assert obj2 == obj
# insure the XASObject is serializable
# import json
# json.dumps(obj2.to_dict())
[docs]
def test_create_from_several_spectrums(tmpdir):
"""check that we can create a XASObject from numpy arrays"""
energy, spectra = example_spectra(shape=(256, 20, 10))
spectra_path = "/data/NXdata/data"
channel_path = "/data/NXdata/Channel"
filename = str(tmpdir / "myfile.h5")
with h5py.File(filename, "a") as f:
f[spectra_path] = spectra
f[channel_path] = energy
xas_obj = read_from_url(
spectra_url=DataUrl(file_path=filename, data_path=spectra_path, scheme="silx"),
channel_url=DataUrl(file_path=filename, data_path=channel_path, scheme="silx"),
dimensions=(2, 1, 0),
)
assert xas_obj.spectra.shape[0] == 20
assert xas_obj.spectra.shape[1] == 10
assert xas_obj.n_spectrum == 20 * 10
ddict = xas_obj.to_dict()
original_spectra = silx.io.utils.get_data(
DataUrl(file_path=filename, data_path=spectra_path, scheme="silx")
)
obj2 = XASObject.from_dict(ddict)
assert xas_obj.n_spectrum == obj2.n_spectrum
obj2_mu_spectra = obj2.spectra.map_to(data_info="mu")
numpy.testing.assert_array_equal(original_spectra, obj2_mu_spectra)
assert obj2 == xas_obj
[docs]
@pytest.fixture()
def serialize_data(tmpdir):
energy, spectra = example_spectra(shape=(256, 20, 10))
spectra_path = "/data/NXdata/data"
channel_path = "/data/NXdata/Channel"
filename = str(tmpdir / "myfile.h5")
with h5py.File(filename, "a") as f:
f[spectra_path] = spectra
f[channel_path] = energy
return energy, spectra, filename
