Does `assert =` work for this case? I would think `assert_allclose` would be better to compare floats?