squirrel.integration_test.shared_fixtures
¶
This module defines shared fixtures that can be used inside a conftest.py like this
from squirrel.integration_test.shared_fixtures import * # noqa: F401, F403
These fixtures will then be shared with the tests under the scope of the conftest.py.
This is useful for splitting unit and integration tests and defining separate conftest files. Specific fixtures for integration or unit tests can then be defined in the respective conftest.
Module Contents¶
Functions¶
|
Set the testing array shape. |
|
Define shards used in testing. |
|
Get the path for testing gcs bucket store. Change of gcs target location should be set here, instead of |
|
Use parametrization in pytest to generate two paths consecutively. Any testing function and fixture |
|
Returns a mock of the pytorch module with parallel and distributed worker. |
|
Pytorch worker config (worker_idx, num_worker). |
|
Pytorch world config (rank_idx, world_size). |