[Idea] Option to add dataflow, model tests

[vfdev-5]

Here is an idea to think about for later versions, like v0.3 etc

Context

Let's imagine I use the generator to quick-start with my specific problem: own dataflow, model etc.
I generate the code and start to bootstrap things between the training code and my custom things. Without running the training, it is almost impossible to ensure the correctness, however, I could think of some basic additional tests with verbose option to ensure that my own dataloaders and the model provide the expected info.

Feature

Let's say generated files are:

- main.py
- model.py
- dataflow.py
- utils.py

The idea is to provide additional folder:

tests
 - test_dataflow.py
 - test_model.py

where we can provide a skeleton code for

• loop over few dataloader batches and either show images (like here)or assert dimensions.
• assert the output shape/type of the model

---

Anyway, this is something to discuss and brain storm...