def test_harmonize_undefined_inchikey(metadata, aliases, undefined, expected):
spectrum_in = SpectrumBuilder().with_metadata(metadata).build()
spectrum = harmonize_undefined_inchikey(spectrum_in,
aliases=aliases,
undefined=undefined)
assert spectrum.get("inchikey") == expected
def test_harmonize_undefined_inchikey_empty_string():
spectrum_in = Spectrum(mz=numpy.array([], dtype="float"),
intensities=numpy.array([], dtype="float"),
metadata={"inchikey": ""})
spectrum = harmonize_undefined_inchikey(spectrum_in)
assert spectrum.get("inchikey") == ""
def test_harmonize_undefined_inchikey_alias_nan_undefined_is_na():
spectrum_in = Spectrum(mz=numpy.array([], dtype="float"),
intensities=numpy.array([], dtype="float"),
metadata={"inchikey": "nan"})
spectrum = harmonize_undefined_inchikey(
spectrum_in, aliases=["nodata", "NaN", "Nan", "nan"], undefined="n/a")
assert spectrum.get("inchikey") == "n/a"
def test_empty_spectrum():
spectrum_in = None
spectrum = harmonize_undefined_inchikey(spectrum_in)
assert spectrum is None, "Expected different handling of None spectrum."