Is your feature request related to a problem? Please describe.
Right now, we don't check the shape of the returned values versus the init's. The length of the inits should match one of the dimensions. This means, that something like scipy handles those errors. However, I think we can be more informative (error-wise) on bad shapes of objects returned by psi.
Describe the solution you'd like
A check on the shape of the returned object. We already extract n via one of the 2D. So, just check the other dimension and assert if it matches len(init).
Will also need to add a quick tests to make sure functions as intended error-wise
pzivich
commented
2 years ago
Is your feature request related to a problem? Please describe. Right now, we don't check the shape of the returned values versus the init's. The length of the inits should match one of the dimensions. This means, that something like
scipyhandles those errors. However, I think we can be more informative (error-wise) on bad shapes of objects returned bypsi.Describe the solution you'd like A check on the shape of the returned object. We already extract
nvia one of the 2D. So, just check the other dimension and assert if it matcheslen(init).Will also need to add a quick tests to make sure functions as intended error-wise
Describe alternatives you've considered None.
Additional context None.