def test_assignment_class(caplog):
for case in subclass_cases:
with caplog.at_level(logging.WARNING):
new_subclass = assign_signal_subclass(
dtype=np.dtype(case.dtype),
signal_dimension=case.sig_dim,
signal_type=case.sig_type,
lazy=False,
)
assert new_subclass is getattr(hs.signals, case.cls)
warn_msg = "not understood. See `hs.print_known_signal_types()` for a list of known signal types"
if case.sig_type == "DefinitelyNotAHyperSpySignal":
assert warn_msg in caplog.text
else:
assert warn_msg not in caplog.text
lazyclass = 'Lazy' + case.cls if case.cls != 'BaseSignal' else 'LazySignal'
new_subclass = assign_signal_subclass(
dtype=np.dtype(case.dtype),
signal_dimension=case.sig_dim,
signal_type=case.sig_type,
lazy=True,
)
assert new_subclass is getattr(_lazy_signals, lazyclass)
def test_assignment_class():
for case in subclass_cases:
assert (assign_signal_subclass(dtype=np.dtype(case.dtype),
signal_dimension=case.sig_dim,
signal_type=case.sig_type,
lazy=False) is getattr(
hs.signals, case.cls))
lazyclass = 'Lazy' + case.cls if case.cls is not 'BaseSignal' \
else 'LazySignal'
assert (assign_signal_subclass(dtype=np.dtype(case.dtype),
signal_dimension=case.sig_dim,
signal_type=case.sig_type,
lazy=True) is getattr(
hs.signals, lazyclass))
def test_assignment_class():
for case in subclass_cases:
assert (
assign_signal_subclass(
dtype=np.dtype(case.dtype),
signal_dimension=case.sig_dim,
signal_type=case.sig_type,
lazy=False) is
getattr(hs.signals, case.cls))
lazyclass = 'Lazy' + case.cls if case.cls is not 'BaseSignal' \
else 'LazySignal'
assert (
assign_signal_subclass(
dtype=np.dtype(case.dtype),
signal_dimension=case.sig_dim,
signal_type=case.sig_type,
lazy=True) is
getattr(_lazy_signals, lazyclass))
def test_image_simulation():
assert_true(assign_signal_subclass(
record_by="image",
signal_type="",
signal_origin="simulation") is ImageSimulation)
def test_eds_tem_spectrum(self):
assert_is(
assign_signal_subclass(dtype=np.dtype('float'),
signal_dimension=1,
signal_type="EDS_TEM"),
hs.signals.EDSTEMSpectrum)
def test_weird_spectrum():
cls = assign_signal_subclass(record_by="spectrum",
signal_type="weird",
signal_origin="")
assert_true(cls is hs.signals.Spectrum)
def test_image_simulation():
assert_true(
assign_signal_subclass(
record_by="image", signal_type="", signal_origin="simulation") is
hs.signals.ImageSimulation)
def test_spectrum():
assert_true(
assign_signal_subclass(record_by="spectrum",
signal_type="",
signal_origin="") is hs.signals.Spectrum)
def test_weird_spectrum():
cls = assign_signal_subclass(
record_by="spectrum",
signal_type="weird",
signal_origin="")
assert_true(cls is Spectrum)
def test_complex_image(self):
assert_is(assign_signal_subclass(
dtype=complex,
signal_dimension=2,
signal_type=""), hs.signals.ComplexSignal2D)
def test_complex_spectrum(self):
assert_is(assign_signal_subclass(
dtype=complex,
signal_dimension=1,
signal_type=""), hs.signals.ComplexSignal1D)
def test_dielectric_function(self):
assert_is(assign_signal_subclass(
dtype=complex,
signal_dimension=1,
signal_type="DielectricFunction"), hs.signals.DielectricFunction)
def test_eels_spectrum(self):
assert_is(assign_signal_subclass(
dtype=np.dtype('float'),
signal_dimension=1,
signal_type="EELS"), hs.signals.EELSSpectrum)
def test_signal2d(self):
assert_is(assign_signal_subclass(
dtype=np.dtype('float'),
signal_dimension=2,
signal_type=""), hs.signals.Signal2D)
def test_eels_spectrum():
assert_true(assign_signal_subclass(
record_by="spectrum",
signal_type="EELS",
signal_origin="") is EELSSpectrum)
def test_eels_spectrum_simulation():
assert_true(assign_signal_subclass(
record_by="spectrum",
signal_type="EELS",
signal_origin="simulation") is
EELSSpectrumSimulation)
def test_weird_real(self):
assert_is(assign_signal_subclass(
dtype=np.dtype('float'),
signal_dimension=1000,
signal_type="weird"), hs.signals.BaseSignal)
def test_signal():
assert_true(assign_signal_subclass(
record_by="",
signal_type="",
signal_origin="") is Signal)
def test_weird_spectrum(self):
assert_is(assign_signal_subclass(
dtype=np.dtype('float'),
signal_dimension=1,
signal_type="weird"), hs.signals.Signal1D)
def test_image():
assert_true(
assign_signal_subclass(record_by="image",
signal_type="",
signal_origin="") is hs.signals.Image)
def test_weird_complex(self):
assert_is(assign_signal_subclass(
dtype=complex,
signal_dimension=1000,
signal_type="weird"), hs.signals.ComplexSignal)
def test_eels_spectrum_simulation():
assert_true(
assign_signal_subclass(record_by="spectrum",
signal_type="EELS",
signal_origin="simulation") is
hs.signals.EELSSpectrumSimulation)
def test_spectrum():
assert_true(assign_signal_subclass(
record_by="spectrum",
signal_type="",
signal_origin="") is hs.signals.Spectrum)
def test_signal():
assert_true(
assign_signal_subclass(record_by="", signal_type="", signal_origin="")
is hs.signals.Signal)
def test_eels_spectrum():
assert_true(
assign_signal_subclass(record_by="spectrum",
signal_type="EELS",
signal_origin="") is EELSSpectrum)
def test_image():
assert_true(assign_signal_subclass(
record_by="image",
signal_type="",
signal_origin="") is Image)
def test_eds_tem_spectrum(self):
assert_is(assign_signal_subclass(
dtype=np.dtype('float'),
signal_dimension=1,
signal_type="EDS_TEM"), hs.signals.EDSTEMSpectrum)
