rename unittests

pyproject.toml

@@ -7,10 +7,10 @@ requires = [
 
 [tool.isort]
 known_first_party = [
-    "docs",
     "pl_examples",
     "pytorch_lightning",
-    "tests",
+    "legacy",
+    "unittests",
 ]
 profile = "black"
 line_length = 120

test/tests/accelerators/test_accelerator_connector.py → test/unittests/accelerators/test_accelerator_connector.py

@@ -45,7 +45,7 @@
     SingleDeviceStrategy,
 )
 from pytorch_lightning.utilities.exceptions import MisconfigurationException
-from tests.helpers.runif import RunIf
+from unittests.helpers.runif import RunIf
 
 
 def test_accelerator_choice_cpu(tmpdir):

test/tests/accelerators/test_cpu.py → test/unittests/accelerators/test_cpu.py

@@ -12,7 +12,7 @@
 from pytorch_lightning.plugins.precision.precision_plugin import PrecisionPlugin
 from pytorch_lightning.strategies import SingleDeviceStrategy
 from pytorch_lightning.utilities.exceptions import MisconfigurationException
-from tests.helpers.boring_model import BoringModel
+from unittests.helpers.boring_model import BoringModel
 
 
 def test_restore_checkpoint_after_pre_setup_default():

test/tests/accelerators/test_gpu.py → test/unittests/accelerators/test_gpu.py

@@ -19,8 +19,8 @@
 from pytorch_lightning import Trainer
 from pytorch_lightning.accelerators import GPUAccelerator
 from pytorch_lightning.accelerators.gpu import get_nvidia_gpu_stats
-from tests.helpers import BoringModel
-from tests.helpers.runif import RunIf
+from unittests.helpers import BoringModel
+from unittests.helpers.runif import RunIf
 
 
 @RunIf(min_cuda_gpus=1)

test/tests/accelerators/test_hpu.py → test/unittests/accelerators/test_hpu.py

@@ -22,10 +22,10 @@
 from pytorch_lightning.strategies.single_hpu import SingleHPUStrategy
 from pytorch_lightning.utilities import _HPU_AVAILABLE
 from pytorch_lightning.utilities.exceptions import MisconfigurationException
-from tests.helpers.boring_model import BoringModel
-from tests.helpers.datamodules import ClassifDataModule
-from tests.helpers.runif import RunIf
-from tests.helpers.simple_models import ClassificationModel
+from unittests.helpers.boring_model import BoringModel
+from unittests.helpers.datamodules import ClassifDataModule
+from unittests.helpers.runif import RunIf
+from unittests.helpers.simple_models import ClassificationModel
 
 
 @RunIf(hpu=True)

test/tests/accelerators/test_ipu.py → test/unittests/accelerators/test_ipu.py

@@ -29,10 +29,10 @@
 from pytorch_lightning.trainer.supporters import CombinedLoader
 from pytorch_lightning.utilities import _IPU_AVAILABLE
 from pytorch_lightning.utilities.exceptions import MisconfigurationException
-from tests.helpers.boring_model import BoringModel
-from tests.helpers.datamodules import ClassifDataModule
-from tests.helpers.runif import RunIf
-from tests.helpers.simple_models import ClassificationModel
+from unittests.helpers.boring_model import BoringModel
+from unittests.helpers.datamodules import ClassifDataModule
+from unittests.helpers.runif import RunIf
+from unittests.helpers.simple_models import ClassificationModel
 
 if _IPU_AVAILABLE:
     import poptorch

test/tests/accelerators/test_tpu.py → test/unittests/accelerators/test_tpu.py

@@ -26,9 +26,9 @@
 from pytorch_lightning.plugins import PrecisionPlugin, TPUPrecisionPlugin, XLACheckpointIO
 from pytorch_lightning.strategies import DDPStrategy, TPUSpawnStrategy
 from pytorch_lightning.utilities import find_shared_parameters
-from tests.helpers.boring_model import BoringModel, RandomDataset
-from tests.helpers.runif import RunIf
-from tests.helpers.utils import pl_multi_process_test
+from unittests.helpers.boring_model import BoringModel, RandomDataset
+from unittests.helpers.runif import RunIf
+from unittests.helpers.utils import pl_multi_process_test
 
 
 class WeightSharingModule(BoringModel):

test/tests/benchmarks/generate_comparison.py → test/unittests/benchmarks/generate_comparison.py

@@ -13,8 +13,8 @@
 # limitations under the License.
 import os
 
-from tests.benchmarks.test_basic_parity import measure_loops
-from tests.helpers.advanced_models import ParityModuleMNIST, ParityModuleRNN
+from unittests.benchmarks.test_basic_parity import measure_loops
+from unittests.helpers.advanced_models import ParityModuleMNIST, ParityModuleRNN
 
 NUM_EPOCHS = 20
 NUM_RUNS = 50

test/tests/benchmarks/test_basic_parity.py → test/unittests/benchmarks/test_basic_parity.py

@@ -21,7 +21,7 @@
 from tqdm import tqdm
 
 from pytorch_lightning import LightningModule, seed_everything, Trainer
-from tests.helpers.advanced_models import ParityModuleCIFAR, ParityModuleMNIST, ParityModuleRNN
+from unittests.helpers.advanced_models import ParityModuleCIFAR, ParityModuleMNIST, ParityModuleRNN
 
 _EXTEND_BENCHMARKS = os.getenv("PL_RUNNING_BENCHMARKS", "0") == "1"
 _SHORT_BENCHMARKS = not _EXTEND_BENCHMARKS

test/tests/benchmarks/test_sharded_parity.py → test/unittests/benchmarks/test_sharded_parity.py

@@ -20,8 +20,8 @@
 
 from pytorch_lightning import seed_everything, Trainer
 from pytorch_lightning.strategies import DDPSpawnShardedStrategy
-from tests.helpers.boring_model import BoringModel, RandomDataset
-from tests.helpers.runif import RunIf
+from unittests.helpers.boring_model import BoringModel, RandomDataset
+from unittests.helpers.runif import RunIf
 
 
 class SeedTrainLoaderModel(BoringModel):

test/tests/benchmarks/test_sync_batchnorm_parity.py → test/unittests/benchmarks/test_sync_batchnorm_parity.py

@@ -17,7 +17,7 @@
 from torch.utils.data import DataLoader, DistributedSampler
 
 from pytorch_lightning import LightningModule, seed_everything, Trainer
-from tests.helpers.runif import RunIf
+from unittests.helpers.runif import RunIf
 
 
 class SyncBNModule(LightningModule):

test/tests/callbacks/test_callback_hook_outputs.py → test/unittests/callbacks/test_callback_hook_outputs.py

@@ -14,7 +14,7 @@
 import pytest
 
 from pytorch_lightning import Callback, Trainer
-from tests.helpers.boring_model import BoringModel
+from unittests.helpers.boring_model import BoringModel
 
 
 @pytest.mark.parametrize("single_cb", [False, True])

test/tests/callbacks/test_callbacks.py → test/unittests/callbacks/test_callbacks.py

@@ -19,8 +19,8 @@
 
 from pytorch_lightning import Callback, Trainer
 from pytorch_lightning.callbacks import ModelCheckpoint
-from tests.helpers import BoringModel
-from tests.helpers.utils import no_warning_call
+from unittests.helpers import BoringModel
+from unittests.helpers.utils import no_warning_call
 
 
 def test_callbacks_configured_in_model(tmpdir):

test/tests/callbacks/test_device_stats_monitor.py → test/unittests/callbacks/test_device_stats_monitor.py

@@ -25,8 +25,8 @@
 from pytorch_lightning.loggers import CSVLogger
 from pytorch_lightning.utilities.exceptions import MisconfigurationException
 from pytorch_lightning.utilities.rank_zero import rank_zero_only
-from tests.helpers import BoringModel
-from tests.helpers.runif import RunIf
+from unittests.helpers import BoringModel
+from unittests.helpers.runif import RunIf
 
 
 @RunIf(min_cuda_gpus=1)

test/tests/callbacks/test_early_stopping.py → test/unittests/callbacks/test_early_stopping.py

@@ -26,10 +26,10 @@
 from pytorch_lightning import seed_everything, Trainer
 from pytorch_lightning.callbacks import EarlyStopping, ModelCheckpoint
 from pytorch_lightning.utilities.exceptions import MisconfigurationException
-from tests.helpers import BoringModel
-from tests.helpers.datamodules import ClassifDataModule
-from tests.helpers.runif import RunIf
-from tests.helpers.simple_models import ClassificationModel
+from unittests.helpers import BoringModel
+from unittests.helpers.datamodules import ClassifDataModule
+from unittests.helpers.runif import RunIf
+from unittests.helpers.simple_models import ClassificationModel
 
 _logger = logging.getLogger(__name__)
 

test/tests/callbacks/test_finetuning_callback.py → test/unittests/callbacks/test_finetuning_callback.py

@@ -22,7 +22,7 @@
 from pytorch_lightning import LightningModule, seed_everything, Trainer
 from pytorch_lightning.callbacks import BackboneFinetuning, BaseFinetuning, ModelCheckpoint
 from pytorch_lightning.utilities.imports import _TORCH_GREATER_EQUAL_1_11
-from tests.helpers import BoringModel, RandomDataset
+from unittests.helpers import BoringModel, RandomDataset
 
 
 class TestBackboneFinetuningCallback(BackboneFinetuning):

test/tests/callbacks/test_gradient_accumulation_scheduler.py → test/unittests/callbacks/test_gradient_accumulation_scheduler.py

@@ -19,7 +19,7 @@
 from pytorch_lightning import Trainer
 from pytorch_lightning.callbacks import GradientAccumulationScheduler
 from pytorch_lightning.utilities.exceptions import MisconfigurationException
-from tests.helpers import BoringModel
+from unittests.helpers import BoringModel
 
 
 @pytest.mark.parametrize("accumulate_grad_batches", (1, 2, 3))

test/tests/callbacks/test_lambda_function.py → test/unittests/callbacks/test_lambda_function.py

@@ -17,8 +17,8 @@
 
 from pytorch_lightning import seed_everything, Trainer
 from pytorch_lightning.callbacks import Callback, LambdaCallback
-from tests.helpers.boring_model import BoringModel
-from tests.models.test_hooks import get_members
+from unittests.helpers.boring_model import BoringModel
+from unittests.models.test_hooks import get_members
 
 
 def test_lambda_call(tmpdir):

test/tests/callbacks/test_lr_monitor.py → test/unittests/callbacks/test_lr_monitor.py

@@ -15,15 +15,15 @@
 import torch
 from torch import optim
 
-import tests.helpers.utils as tutils
+import unittests.helpers.utils as tutils
 from pytorch_lightning import Trainer
 from pytorch_lightning.callbacks import LearningRateMonitor
 from pytorch_lightning.callbacks.callback import Callback
 from pytorch_lightning.callbacks.finetuning import BackboneFinetuning
 from pytorch_lightning.utilities.exceptions import MisconfigurationException
-from tests.helpers import BoringModel
-from tests.helpers.datamodules import ClassifDataModule
-from tests.helpers.simple_models import ClassificationModel
+from unittests.helpers import BoringModel
+from unittests.helpers.datamodules import ClassifDataModule
+from unittests.helpers.simple_models import ClassificationModel
 
 
 def test_lr_monitor_single_lr(tmpdir):

test/tests/callbacks/test_model_summary.py → test/unittests/callbacks/test_model_summary.py

@@ -15,7 +15,7 @@
 
 from pytorch_lightning import Trainer
 from pytorch_lightning.callbacks import ModelSummary
-from tests.helpers.boring_model import BoringModel
+from unittests.helpers.boring_model import BoringModel
 
 
 def test_model_summary_callback_present_trainer():

test/tests/callbacks/test_prediction_writer.py → test/unittests/callbacks/test_prediction_writer.py

@@ -21,8 +21,8 @@
 from pytorch_lightning.callbacks import BasePredictionWriter
 from pytorch_lightning.trainer.supporters import CombinedLoader
 from pytorch_lightning.utilities.exceptions import MisconfigurationException
-from tests.helpers import BoringModel, RandomDataset
-from tests.helpers.runif import RunIf
+from unittests.helpers import BoringModel, RandomDataset
+from unittests.helpers.runif import RunIf
 
 
 class DummyPredictionWriter(BasePredictionWriter):

test/tests/callbacks/test_pruning.py → test/unittests/callbacks/test_pruning.py

@@ -25,8 +25,8 @@
 from pytorch_lightning import Trainer
 from pytorch_lightning.callbacks import ModelCheckpoint, ModelPruning
 from pytorch_lightning.utilities.exceptions import MisconfigurationException
-from tests.helpers.boring_model import BoringModel
-from tests.helpers.runif import RunIf
+from unittests.helpers.boring_model import BoringModel
+from unittests.helpers.runif import RunIf
 
 
 class TestModel(BoringModel):

test/tests/callbacks/test_quantization.py → test/unittests/callbacks/test_quantization.py

@@ -24,10 +24,10 @@
 from pytorch_lightning.callbacks import QuantizationAwareTraining
 from pytorch_lightning.utilities.exceptions import MisconfigurationException
 from pytorch_lightning.utilities.memory import get_model_size_mb
-from tests.helpers.boring_model import RandomDataset
-from tests.helpers.datamodules import RegressDataModule
-from tests.helpers.runif import RunIf
-from tests.helpers.simple_models import RegressionModel
+from unittests.helpers.boring_model import RandomDataset
+from unittests.helpers.datamodules import RegressDataModule
+from unittests.helpers.runif import RunIf
+from unittests.helpers.simple_models import RegressionModel
 
 
 @pytest.mark.parametrize("observe", ["average", "histogram"])

test/tests/callbacks/test_rich_model_summary.py → test/unittests/callbacks/test_rich_model_summary.py

@@ -20,8 +20,8 @@
 from pytorch_lightning import Trainer
 from pytorch_lightning.callbacks import RichModelSummary, RichProgressBar
 from pytorch_lightning.utilities.model_summary import summarize
-from tests.helpers import BoringModel
-from tests.helpers.runif import RunIf
+from unittests.helpers import BoringModel
+from unittests.helpers.runif import RunIf
 
 
 @RunIf(rich=True)

test/tests/callbacks/test_rich_progress_bar.py → test/unittests/callbacks/test_rich_progress_bar.py

@@ -21,8 +21,8 @@
 from pytorch_lightning import Trainer
 from pytorch_lightning.callbacks import ProgressBarBase, RichProgressBar
 from pytorch_lightning.callbacks.progress.rich_progress import RichProgressBarTheme
-from tests.helpers.boring_model import BoringModel, RandomDataset, RandomIterableDataset
-from tests.helpers.runif import RunIf
+from unittests.helpers.boring_model import BoringModel, RandomDataset, RandomIterableDataset
+from unittests.helpers.runif import RunIf
 
 
 @RunIf(rich=True)

test/tests/callbacks/test_stochastic_weight_avg.py → test/unittests/callbacks/test_stochastic_weight_avg.py

@@ -24,8 +24,8 @@
 from pytorch_lightning.callbacks import StochasticWeightAveraging
 from pytorch_lightning.strategies import DDPSpawnStrategy, Strategy
 from pytorch_lightning.utilities.exceptions import MisconfigurationException
-from tests.helpers.boring_model import BoringModel, RandomDataset, RandomIterableDataset
-from tests.helpers.runif import RunIf
+from unittests.helpers.boring_model import BoringModel, RandomDataset, RandomIterableDataset
+from unittests.helpers.runif import RunIf
 
 
 class SwaTestModel(BoringModel):

test/tests/callbacks/test_timer.py → test/unittests/callbacks/test_timer.py

@@ -22,8 +22,8 @@
 from pytorch_lightning.callbacks import ModelCheckpoint
 from pytorch_lightning.callbacks.timer import Timer
 from pytorch_lightning.utilities.exceptions import MisconfigurationException
-from tests.helpers import BoringModel
-from tests.helpers.runif import RunIf
+from unittests.helpers import BoringModel
+from unittests.helpers.runif import RunIf
 
 
 def test_trainer_flag(caplog):

test/tests/callbacks/test_tqdm_progress_bar.py → test/unittests/callbacks/test_tqdm_progress_bar.py

@@ -29,8 +29,8 @@
 from pytorch_lightning.callbacks.progress.tqdm_progress import Tqdm
 from pytorch_lightning.core.module import LightningModule
 from pytorch_lightning.utilities.exceptions import MisconfigurationException
-from tests.helpers.boring_model import BoringModel, RandomDataset
-from tests.helpers.runif import RunIf
+from unittests.helpers.boring_model import BoringModel, RandomDataset
+from unittests.helpers.runif import RunIf
 
 
 class MockTqdm(Tqdm):

test/tests/checkpointing/test_checkpoint_callback_frequency.py → test/unittests/checkpointing/test_checkpoint_callback_frequency.py

@@ -18,8 +18,8 @@
 import torch
 
 from pytorch_lightning import callbacks, Trainer
-from tests.helpers import BoringModel
-from tests.helpers.runif import RunIf
+from unittests.helpers import BoringModel
+from unittests.helpers.runif import RunIf
 
 
 def test_disabled_checkpointing(tmpdir):

test/tests/checkpointing/test_legacy_checkpoints.py → test/unittests/checkpointing/test_legacy_checkpoints.py

@@ -22,7 +22,7 @@
 
 import pytorch_lightning as pl
 from pytorch_lightning import Callback, Trainer
-from tests import _PATH_LEGACY, _PROJECT_ROOT
+from unittests import _PATH_LEGACY, _PROJECT_ROOT
 
 LEGACY_CHECKPOINTS_PATH = os.path.join(_PATH_LEGACY, "checkpoints")
 CHECKPOINT_EXTENSION = ".ckpt"

test/tests/checkpointing/test_model_checkpoint.py → test/unittests/checkpointing/test_model_checkpoint.py

@@ -31,15 +31,15 @@
 from torch import optim
 
 import pytorch_lightning as pl
-import tests.helpers.utils as tutils
+import unittests.helpers.utils as tutils
 from pytorch_lightning import seed_everything, Trainer
 from pytorch_lightning.callbacks import ModelCheckpoint
 from pytorch_lightning.loggers import TensorBoardLogger
 from pytorch_lightning.utilities.cloud_io import load as pl_load
 from pytorch_lightning.utilities.exceptions import MisconfigurationException
 from pytorch_lightning.utilities.imports import _OMEGACONF_AVAILABLE
-from tests.helpers import BoringModel
-from tests.helpers.runif import RunIf
+from unittests.helpers import BoringModel
+from unittests.helpers.runif import RunIf
 
 if _OMEGACONF_AVAILABLE:
     from omegaconf import Container, OmegaConf

test/tests/checkpointing/test_torch_saving.py → test/unittests/checkpointing/test_torch_saving.py

@@ -16,8 +16,8 @@
 import torch
 
 from pytorch_lightning import Trainer
-from tests.helpers import BoringModel
-from tests.helpers.runif import RunIf
+from unittests.helpers import BoringModel
+from unittests.helpers.runif import RunIf
 
 
 def test_model_torch_save(tmpdir):

test/tests/checkpointing/test_trainer_checkpoint.py → test/unittests/checkpointing/test_trainer_checkpoint.py

@@ -23,7 +23,7 @@
 from pytorch_lightning.callbacks import ModelCheckpoint
 from pytorch_lightning.plugins.io.torch_plugin import TorchCheckpointIO
 from pytorch_lightning.plugins.io.xla_plugin import XLACheckpointIO
-from tests.helpers import BoringModel
+from unittests.helpers import BoringModel
 
 
 def test_finetuning_with_ckpt_path(tmpdir):

test/tests/conftest.py → test/unittests/conftest.py

@@ -24,7 +24,7 @@
 from pytorch_lightning.plugins.environments.lightning_environment import find_free_network_port
 from pytorch_lightning.trainer.connectors.signal_connector import SignalConnector
 from pytorch_lightning.utilities.imports import _IS_WINDOWS
-from tests import _PATH_DATASETS
+from unittests import _PATH_DATASETS
 
 
 @pytest.fixture(scope="session")