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 run(parakeet_fn, python_fn): testing_helpers.expect_allpairs(jit(parakeet_fn), python_fn, values)
def test_adverb_outer_prod(): expect_allpairs(adverb_outer_prod, np.multiply.outer, vectors)
def test_loopdot(): expect_allpairs(loop_dot, np.dot, 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)
def test_adverb_outer_prod():
expect_allpairs(adverb_outer_prod, np.multiply.outer, vectors)
