This should probably use ``allclose`` or one of those Numpy functions, rather than relying on floating-point equality.