Add policy tests for learned dynamics.

[hjsuh94]

This fixes some bugs in the dynamics training and tests that the policies all work successfully when dynamics is a NNDynamicsSystem.

This change is