def test_allpairs_elt_diff(): def python_impl(x,y): nx = len(x) ny = len(y) result = np.zeros(shape = (nx,ny), dtype=(x[0]-y[0]).dtype) for i in xrange(nx): for j in xrange(ny): result[i,j] = x[i] - y[j] return result expect_allpairs(allpairs_elt_diff, python_impl, vecs)
def test_dot(): expect_allpairs(dot, np.dot, vecs)
def test_adverb_matmult(): expect_allpairs(adverb_matmult, lambda x, y: np.dot(x, y.T), matrices)
def test_loopdot(): expect_allpairs(loop_dot, np.dot, vectors)
def test_adverb_dot(): expect_allpairs(dot, lambda x, y: np.sum(x * y), vectors)
def run(parakeet_fn, python_fn): testing_helpers.expect_allpairs(jit(parakeet_fn), python_fn, values)
def test_conditional_div(): expect_allpairs(each_conditional_div, python_conditional_div, [ints_1d, floats_1d])
def test_implicit_add_mat(): expect_allpairs(add, np.add, matrices)
def test_explicit_add_vec(): expect_allpairs(each_add, np.add, vecs)
def test_adverb_matmult(): expect_allpairs(adverb_matmult, transposed_np_dot, matrices)
def test_implicit_add_vec(): expect_allpairs(add, np.add, vecs)
def test_dot(): vecs = [m[0] for m in matrices] expect_allpairs(dot, np.dot, vecs)
def test_adverb_outer_prod(): expect_allpairs(adverb_outer_prod, np.multiply.outer, vectors)
def test_explicit_add_mat(): expect_allpairs(each_add, np.add, matrices)
def test_adverb_dot(): expect_allpairs(dot, lambda x,y: np.sum(x*y), vectors)