tnmf.tests.test_backends

Test the decomposition on two identical images and ensure that all backends yield the same factorization.

Module Contents

Functions

fit_nmf(backend, reconstruction_mode)
fixture_expected_factorization(reconstruction_mode)
test_expected_energy(backend: str, reconstruction_mode: str, expected_factorization: Tuple[numpy.ndarray, numpy.ndarray])

Attributes

expected_energies
backends
reconstruction_modes
raise_not_implemented_errors
img
V

tnmf.tests.test_backends.expected_energies

tnmf.tests.test_backends.backends = ['numpy', 'numpy_fft', 'numpy_caching_fft', 'pytorch', 'pytorch_fft']

tnmf.tests.test_backends.reconstruction_modes

tnmf.tests.test_backends.raise_not_implemented_errors = False

tnmf.tests.test_backends.img

tnmf.tests.test_backends.V

tnmf.tests.test_backends.fit_nmf(backend, reconstruction_mode)

tnmf.tests.test_backends.fixture_expected_factorization(reconstruction_mode)

tnmf.tests.test_backends.test_expected_energy(backend: str, reconstruction_mode: str, expected_factorization: Tuple[numpy.ndarray, numpy.ndarray])