feat(app): add an option to include test file (#97)

* Merge dev to master for v0.1.0 release (#94)

* Making Text classification template similar to Image Classification (#92)

* Fix template sidebar

* Code style

* [skip ci] updated contributing and readme (#93)

* [skip ci] updated contributing

* [skip ci] Updated readme

Co-authored-by: Taras Savchyn <30748114+trsvchn@users.noreply.github.com>

* feat(app): add an option to include test file

* fix: test_all in generate, pytest as optional dep

Co-authored-by: vfdev <vfdev.5@gmail.com>
Co-authored-by: Taras Savchyn <30748114+trsvchn@users.noreply.github.com>

Author: 3 people

app/codegen.py

@@ -28,7 +28,13 @@ def render_templates(self, template_name: str, config: dict):
             trim_blocks=True,
             lstrip_blocks=True,
         )
-        for fname in env.list_templates(filter_func=lambda x: not x.startswith("_")):
+
+        def filter_func(x: str):
+            if config["test_all"]:
+                return not x.startswith("_")
+            return not x.startswith("_") ^ (x == "test_all.py")
+
+        for fname in env.list_templates(filter_func=filter_func):
             code = env.get_template(fname).render(**config)
             self.rendered_code[fname] = code
             yield fname, code

templates/_base/_sidebar.py

@@ -106,6 +106,16 @@ def ignite_loggers_options(config):
     st.markdown("---")
 
 
+def test_all_options(config):
+    st.markdown("## Unit Test Options")
+    config["test_all"] = st.checkbox(
+        "Include a test file for the generated codes", help="Tests are implemented with pytest."
+    )
+    if config["test_all"]:
+        config["test_deps"] = "pytest"
+    st.markdown("---")
+
+
 def _setup_common_training_handlers_options(config):
     config["save_every_iters"] = st.number_input(
         "Saving iteration interval (save_every_iters)", min_value=1, value=1000

templates/gan/_sidebar.py

@@ -9,6 +9,7 @@
     distributed_options,
     ignite_handlers_options,
     ignite_loggers_options,
+    test_all_options,
 )
 
 
@@ -79,5 +80,6 @@ def get_configs() -> dict:
         distributed_options(config)
         ignite_handlers_options(config)
         ignite_loggers_options(config)
+        test_all_options(config)
 
     return config

templates/gan/requirements.txt

@@ -3,6 +3,6 @@ pytorch-ignite>=0.4.4
 torchvision>=0.8.0
 matplotlib>=3.3.0
 pandas
-pytest
+{{ test_deps }}
 {{ handler_deps }}
 {{ logger_deps }}

templates/image_classification/_sidebar.py

@@ -9,6 +9,7 @@
     distributed_options,
     ignite_handlers_options,
     ignite_loggers_options,
+    test_all_options,
 )
 
 
@@ -95,5 +96,6 @@ def get_configs():
         distributed_options(config)
         ignite_handlers_options(config)
         ignite_loggers_options(config)
+        test_all_options(config)
 
     return config

templates/image_classification/requirements.txt

@@ -1,5 +1,5 @@
 torch>=1.7.0
 pytorch-ignite>=0.4.4
-pytest
+{{ test_deps }}
 {{ handler_deps }}
 {{ logger_deps }}

templates/image_classification/trainers.py

@@ -5,7 +5,6 @@
 
 import torch
 from ignite.engine import Engine
-from ignite.metrics import loss
 from torch.cuda.amp import autocast
 from torch.optim.optimizer import Optimizer
 

templates/single/_sidebar.py

@@ -9,6 +9,7 @@
     distributed_options,
     ignite_handlers_options,
     ignite_loggers_options,
+    test_all_options,
 )
 
 
@@ -25,5 +26,6 @@ def get_configs():
         distributed_options(config)
         ignite_handlers_options(config)
         ignite_loggers_options(config)
+        test_all_options(config)
 
     return config

templates/single/requirements.txt

@@ -1,6 +1,6 @@
 setuptools
 torch>=1.7.0
 pytorch-ignite>=0.4.4
-pytest
+{{ test_deps }}
 {{ handler_deps }}
 {{ logger_deps }}

templates/text_classification/requirements.txt

@@ -2,5 +2,6 @@ torch>=1.7.0
 pytorch-ignite>=0.4.4
 transformers
 datasets
+{{ test_deps }}
 {{ handler_deps }}
 {{ logger_deps }}