Module neuralops.tests.test_all_compressor_modules

Smoke tests for ALL registered compressor modules.

Functions

def test_compressor_module_importable(module_name)

Expand source code

@pytest.mark.smoke
@pytest.mark.parametrize("module_name", sorted(COMPRESSOR_REGISTRY.keys()))
def test_compressor_module_importable(module_name):
    """Verify that every registered compressor key can be retrieved."""
    with warnings.catch_warnings():
        warnings.simplefilter("ignore")
        cls = COMPRESSOR_REGISTRY.get(module_name)
    assert cls is not None

Verify that every registered compressor key can be retrieved.