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)