TST: assert_equal -> assert_allclose