sparsecoding.test_utils package

sparsecoding.test_utils.assert_allclose(a: ndarray, b: ndarray, rtol: float = 1e-05, atol: float = 1e-06) → None

sparsecoding.test_utils.assert_shape_equal(a: ndarray, b: ndarray) → None

sparsecoding.test_utils.bars_datas_fixture(patch_size_fixture: int, dataset_size_fixture: int, bars_datasets_fixture: list[BarsDataset]) → list[Tensor]

sparsecoding.test_utils.bars_datasets_fixture(patch_size_fixture: int, dataset_size_fixture: int, priors_fixture: list[Prior]) → list[BarsDataset]

sparsecoding.test_utils.bars_dictionary_fixture(patch_size_fixture: int, bars_datasets_fixture: list[BarsDataset]) → Tensor: Return a bars dataset basis reshaped to represent a dictionary.

sparsecoding.test_utils.dataset_size_fixture() → int

sparsecoding.test_utils.patch_size_fixture() → int

sparsecoding.test_utils.priors_fixture(patch_size_fixture: int) → list[Prior]

Submodules

sparsecoding.test_utils.asserts module

sparsecoding.test_utils.asserts.assert_allclose(a: ndarray, b: ndarray, rtol: float = 1e-05, atol: float = 1e-06) → None

sparsecoding.test_utils.asserts.assert_shape_equal(a: ndarray, b: ndarray) → None

sparsecoding.test_utils.asserts_test module

sparsecoding.test_utils.asserts_test.test_assert_np_shape_equal()

sparsecoding.test_utils.asserts_test.test_assert_pytorch_shape_equal()

sparsecoding.test_utils.asserts_test.test_np_all_close()

sparsecoding.test_utils.asserts_test.test_pytorch_all_close()

sparsecoding.test_utils.constant_fixtures module

sparsecoding.test_utils.constant_fixtures.dataset_size_fixture() → int

sparsecoding.test_utils.constant_fixtures.patch_size_fixture() → int

sparsecoding.test_utils.dataset_fixtures module

sparsecoding.test_utils.dataset_fixtures.bars_datas_fixture(patch_size_fixture: int, dataset_size_fixture: int, bars_datasets_fixture: list[BarsDataset]) → list[Tensor]

sparsecoding.test_utils.dataset_fixtures.bars_datasets_fixture(patch_size_fixture: int, dataset_size_fixture: int, priors_fixture: list[Prior]) → list[BarsDataset]

sparsecoding.test_utils.model_fixtures module

sparsecoding.test_utils.model_fixtures.bars_dictionary_fixture(patch_size_fixture: int, bars_datasets_fixture: list[BarsDataset]) → Tensor: Return a bars dataset basis reshaped to represent a dictionary.

sparsecoding.test_utils.prior_fixtures module

sparsecoding.test_utils.prior_fixtures.priors_fixture(patch_size_fixture: int) → list[Prior]