289ae46ae4b3e1abbf135e204c62908aeadc7d85,scipy/spatial/tests/test_kdtree.py,Test_vectorization,test_vectorized_query_all_neighbors,#Test_vectorization#,216

Before Change

        assert_(isinstance(i, list))

    def test_vectorized_query_all_neighbors(self):
        d, i = self.kdtree.query(np.zeros((2, 4, 3)), k=None, distance_upper_bound=1.1)
        assert_equal(np.shape(d), (2, 4))
        assert_equal(np.shape(i), (2, 4))

After Change

    def test_vectorized_query_all_neighbors(self):
        query_shape = (2, 4)
        r = 1.1
        np.random.seed(1234)
        points = np.random.rand(*query_shape, self.kdtree.m)
        d, i = self.kdtree.query(points, k=None, distance_upper_bound=r)
        assert_equal(np.shape(d), query_shape)
        assert_equal(np.shape(i), query_shape)

Instances