To be added to the model unit tests. Adapt BatchingNet and MultiIONet in a similar manner to that in which SimpleNet is already used.