TST use assert_allclose instead of _array_equal