@ellisonbg I'm a little confused by the matplotlib ex03. One of the asserts requires that psi.shape=(10,), but then later the contour requires that psi.shape=(10,10). I'm not sure how to resolve this. Am I missing something?
(I'm getting an error that contour must take a 2D shape for z, which makes sense, but the assert statement says otherwise.)