combustion.testing¶
Extensions to torch.nn
, ranging from fundamental layers
up to larger building blocks.
combustion.testing
Assertions¶
PyTest Decorators¶
LightningModuleTest¶
In order to facilitate model testing without excessive boilerplate code,
a pytest base class is provided that attempts to provide a set of minimal
tests for a PyTorch Lightning LightningModule
. By implementing a
small number of fixtures that provide a model to be tested and data
that can be used for testing, many stages of the LightningModule
lifecycle can be tested without writing any additional test code.