I think you meant to use ``assert_allclose`` here?