From 3ad7511acad22536bd8af6a0d5f5fc17966564a6 Mon Sep 17 00:00:00 2001 From: "sweep-nightly[bot]" <131841235+sweep-nightly[bot]@users.noreply.github.com> Date: Sat, 21 Oct 2023 22:09:54 +0000 Subject: [PATCH 1/3] feat: add tests for data loading, preprocessing, a --- src/test_main.py | 25 +++++++++++++++++++++++++ 1 file changed, 25 insertions(+) create mode 100644 src/test_main.py diff --git a/src/test_main.py b/src/test_main.py new file mode 100644 index 0000000..a9fe569 --- /dev/null +++ b/src/test_main.py @@ -0,0 +1,25 @@ +import pytest +from unittest.mock import Mock, patch +from main import transform, trainset, trainloader, Net + +def test_data_loading_and_preprocessing(): + """ + Test the data loading and preprocessing steps. + This test asserts that the correct transformations are applied to the dataset and that the DataLoader is created with the correct parameters. + """ + with patch('main.datasets.MNIST', new=Mock()) as mock_dataset, patch('main.DataLoader', new=Mock()) as mock_dataloader: + assert mock_dataset.called + assert mock_dataset.call_args[1]['transform'] == transform + assert mock_dataloader.called + assert mock_dataloader.call_args[0][0] == trainset + assert mock_dataloader.call_args[1]['batch_size'] == 64 + assert mock_dataloader.call_args[1]['shuffle'] == True + +def test_model_definition(): + """ + Test the model definition. + This test asserts that the model is defined correctly. + """ + with patch('main.Net', new=Mock()) as mock_model: + assert mock_model.called + assert isinstance(mock_model.return_value, Net) From 3cf229ed203da2e634d5e0efb1a43c39c8d5d4df Mon Sep 17 00:00:00 2001 From: "sweep-nightly[bot]" <131841235+sweep-nightly[bot]@users.noreply.github.com> Date: Sat, 21 Oct 2023 22:17:38 +0000 Subject: [PATCH 2/3] feat: Updated src/test_main.py --- src/test_main.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/src/test_main.py b/src/test_main.py index a9fe569..5af4a22 100644 --- a/src/test_main.py +++ b/src/test_main.py @@ -8,12 +8,12 @@ def test_data_loading_and_preprocessing(): This test asserts that the correct transformations are applied to the dataset and that the DataLoader is created with the correct parameters. """ with patch('main.datasets.MNIST', new=Mock()) as mock_dataset, patch('main.DataLoader', new=Mock()) as mock_dataloader: - assert mock_dataset.called - assert mock_dataset.call_args[1]['transform'] == transform - assert mock_dataloader.called - assert mock_dataloader.call_args[0][0] == trainset - assert mock_dataloader.call_args[1]['batch_size'] == 64 - assert mock_dataloader.call_args[1]['shuffle'] == True + pytest.assertTrue(mock_dataset.called) + pytest.assertEqual(mock_dataset.call_args[1]['transform'], transform) + pytest.assertTrue(mock_dataloader.called) + pytest.assertEqual(mock_dataloader.call_args[0][0], trainset) + pytest.assertEqual(mock_dataloader.call_args[1]['batch_size'], 64) + pytest.assertTrue(mock_dataloader.call_args[1]['shuffle']) def test_model_definition(): """ @@ -21,5 +21,5 @@ def test_model_definition(): This test asserts that the model is defined correctly. """ with patch('main.Net', new=Mock()) as mock_model: - assert mock_model.called - assert isinstance(mock_model.return_value, Net) + pytest.assertTrue(mock_model.called) + pytest.assertIsInstance(mock_model.return_value, Net) From 056cf621de9f058ad383ab5a500d082631c6b902 Mon Sep 17 00:00:00 2001 From: "sweep-nightly[bot]" <131841235+sweep-nightly[bot]@users.noreply.github.com> Date: Sat, 21 Oct 2023 22:18:25 +0000 Subject: [PATCH 3/3] Sandbox run src/test_main.py --- src/test_main.py | 20 +++++++++++++------- 1 file changed, 13 insertions(+), 7 deletions(-) diff --git a/src/test_main.py b/src/test_main.py index 5af4a22..872ce04 100644 --- a/src/test_main.py +++ b/src/test_main.py @@ -1,25 +1,31 @@ -import pytest from unittest.mock import Mock, patch -from main import transform, trainset, trainloader, Net + +import pytest + +from main import Net, trainset, transform + def test_data_loading_and_preprocessing(): """ Test the data loading and preprocessing steps. This test asserts that the correct transformations are applied to the dataset and that the DataLoader is created with the correct parameters. """ - with patch('main.datasets.MNIST', new=Mock()) as mock_dataset, patch('main.DataLoader', new=Mock()) as mock_dataloader: + with patch("main.datasets.MNIST", new=Mock()) as mock_dataset, patch( + "main.DataLoader", new=Mock() + ) as mock_dataloader: pytest.assertTrue(mock_dataset.called) - pytest.assertEqual(mock_dataset.call_args[1]['transform'], transform) + pytest.assertEqual(mock_dataset.call_args[1]["transform"], transform) pytest.assertTrue(mock_dataloader.called) pytest.assertEqual(mock_dataloader.call_args[0][0], trainset) - pytest.assertEqual(mock_dataloader.call_args[1]['batch_size'], 64) - pytest.assertTrue(mock_dataloader.call_args[1]['shuffle']) + pytest.assertEqual(mock_dataloader.call_args[1]["batch_size"], 64) + pytest.assertTrue(mock_dataloader.call_args[1]["shuffle"]) + def test_model_definition(): """ Test the model definition. This test asserts that the model is defined correctly. """ - with patch('main.Net', new=Mock()) as mock_model: + with patch("main.Net", new=Mock()) as mock_model: pytest.assertTrue(mock_model.called) pytest.assertIsInstance(mock_model.return_value, Net)