Now using assert_array_almost_equal