datamodule_test

datamodule tests.

cneuromax.projects.classify_mnist.datamodule_test.datamodule(tmp_path)

MNISTDataModule fixture.

Parameters:

tmp_path (Path) – The temporary path for the MNISTDataModule.

Return type:

MNISTDataModule

Returns:

A generic MNISTDataModule instance.

cneuromax.projects.classify_mnist.datamodule_test.test_setup_fit(datamodule)

Tests setup() #1.

Verifies that setup() behaves correctly when stage is "fit".

Parameters:

datamodule (MNISTDataModule) – A generic MNISTDataModule instance, see datamodule().

Return type:

None

cneuromax.projects.classify_mnist.datamodule_test.test_setup_test(datamodule)

Tests setup() #2.

Verifies that setup() behaves correctly when stage is "test".

Parameters:

datamodule (MNISTDataModule) – A generic MNISTDataModule instance, see datamodule().

Return type:

None