Add test for running model

[ajtritt]

Add a test for running a forward pass on the model.

• ONNX can fix dimension of the model input. We should add a test to make sure the dimension we want to be dynamic are in fact dynamic.