def test_query_linear(self): data_local = [(1, array([1.0, 2.0, 3.0])), (2, array([2.0, 2.0, 4.0])), (3, array([4.0, 2.0, 1.0]))] data = self.sc.parallelize(data_local) inds = array([array([1, 2]), array([3])]) ts = query(data, inds) assert (allclose(ts[0, :], array([1.5, 2., 3.5]))) assert (allclose(ts[1, :], array([4.0, 2.0, 1.0])))
def test_query_linear(self): data_local = [ (1, array([1.0, 2.0, 3.0])), (2, array([2.0, 2.0, 4.0])), (3, array([4.0, 2.0, 1.0])) ] data = self.sc.parallelize(data_local) inds = array([array([1, 2]), array([3])]) ts = query(data, inds) assert(allclose(ts[0, :], array([1.5, 2., 3.5]))) assert(allclose(ts[1, :], array([4.0, 2.0, 1.0])))