Пример #1
0
    def test_set_performance_batch_size_on_empty_config(self) -> None:
        """Test set_performance_batch_size."""
        config = Config()

        config.set_performance_batch_size(1234)

        self.assertIsNone(config.evaluation)
Пример #2
0
    def test_set_performance_batch_size(self) -> None:
        """Test set_performance_batch_size."""
        config = Config(self.predefined_config)

        config.set_performance_batch_size(1234)

        self.assertEqual(1234,
                         config.evaluation.performance.dataloader.batch_size)