TST: modified the tests to change it to assert_array_almost_equal