Generalize check_1d_shape to check_shape

[btjanaka]

Description

There is currently a check called check_1d_shape; we can instead make it a general check_shape function to check arbitrary shapes.

TODO

• [x] Generalize function
• [x] Fix calls

Questions

Status

• [x] I have read the guidelines in CONTRIBUTING.md
• [x] I have formatted my code using yapf
• [x] I have tested my code by running pytest
• [x] I have linted my code with pylint
• [N/A] I have added a one-line description of my change to the changelog in HISTORY.md
• [x] This PR is ready to go