Ensure that you can match test data to the input shape of a neural network