These tests aim to make sure the models run the forward correctly with a given mock data.
These tests aim to make sure the models run the forward correctly with a given mock data.