add epochs smoke test:

Author: nerdai

tests/smoke_tests/ditto_config_epochs.yaml

@@ -0,0 +1,10 @@
+# Parameters that describe server
+n_server_rounds: 1 # The number of rounds to run FL
+
+# Parameters that describe clients
+n_clients: 2 # The number of clients in the FL experiment
+local_epochs: 1 # The number of epochs to complete for client
+batch_size: 32 # The batch size for client training
+
+# Downsampling settings per client
+downsampling_ratio: 0.1 # percentage of original MNIST data to keep for minority numbers

tests/smoke_tests/test_smoke_tests.py

@@ -307,6 +307,19 @@ async def test_ditto_mnist_dynamic() -> None:
     assert_on_done_task(task)
 
 
+@pytest.mark.smoketest
+async def test_ditto_mnist_dynamic_train_by_epochs() -> None:
+    coroutine = run_smoke_test(
+        server_python_path="examples.ditto_example.server",
+        client_python_path="examples.ditto_example.client_dynamic",
+        config_path="tests/smoke_tests/ditto_config_epochs.yaml",
+        dataset_path="examples/datasets/mnist_data/",
+    )
+    task = asyncio.create_task(coroutine)
+    await try_running_test_task(task)
+    assert_on_done_task(task)
+
+
 @pytest.mark.smoketest
 async def test_mr_mtl_mnist(tolerance: float) -> None:
     coroutine = run_smoke_test(