def testShiftZip(self):
self.assertTrue(
recursive_equal(shift_zip(self.x, self.y, (-1, -2), 1),
[(2, 4), (3, 5), (4, 6), (5, 7), (-1, -2)]))
self.assertTrue(
recursive_equal(shift_zip(self.x, self.y, (-3, -4), -1),
[(-3, -4), (1, 3), (2, 4), (3, 5), (4, 6)]))
def testShiftZip(self):
self.assertTrue(
recursive_equal(shift_zip(self.x, self.y, (-1, -2), 1),
[(2, 4), (3, 5), (4, 6), (5, 7), (-1, -2)]))
self.assertTrue(
recursive_equal(shift_zip(self.x, self.y, (-3, -4), -1), [(-3, -4),
(1, 3),
(2, 4),
(3, 5),
(4, 6)]))
def testUnequalTriplyNested(self):
a = [[[1, 2], [3, 4, 5]],
[[6, 7, 8, 9], [10, 11, 12, 13, 14], [15, 16, 17, 18, 19, 20]]]
b = [[[1, 2], [3, 4, 5]],
[[6, 7, 8, 9, 10], [10, 11, 12, 13, 14], [15, 16, 17, 18, 19,
20]]]
self.assertFalse(recursive_equal(a, cuarray(b)))
def testUnequalTriplyNested(self):
a = [[[1,2], [3,4,5]],
[[6,7,8,9], [10,11,12,13,14],
[15,16,17,18,19,20]]]
b = [[[1,2], [3,4,5]],
[[6,7,8,9,10], [10,11,12,13,14],
[15,16,17,18,19,20]]]
self.assertFalse(recursive_equal(a, cuarray(b)))
def testNumpyNested(self):
a = [[np.array([1, 2]), np.array([3, 4, 5])],
[
np.array([6, 7, 8, 9]),
np.array([10, 11, 12, 13, 14]),
np.array([15, 16, 17, 18, 19, 20])
]]
self.assertTrue(recursive_equal(a, cuarray(a)))
def testSub(self):
source = [1, 2, 3, 4, 5]
result = test_sub(source)
self.assertTrue(recursive_equal([5, 4, 3, 2, 1], result))
def testReplNestedTuple(self):
a = test_repl(((1, 2), 3), 2)
self.assertTrue(recursive_equal(a, [((1, 2), 3), ((1, 2), 3)]))
def testUnzip(self):
self.assertTrue(
recursive_equal(test_unzip(test_zip(self.x, self.y)),
([1, 2, 3, 4, 5], [3, 4, 5, 6, 7])))
def testNumpyNested(self):
a = [[np.array([1,2]), np.array([3,4,5])],
[np.array([6,7,8,9]), np.array([10,11,12,13,14]),
np.array([15,16,17,18,19,20])]]
self.assertTrue(recursive_equal(a, cuarray(a)))
def testNumpyFlat(self):
a = np.array([1,2,3,4,5])
self.assertTrue(recursive_equal(a, cuarray(a)))
def testTupleSeqArgs(self):
self.assertTrue(recursive_equal(test_tuple_seq_args(([1,2],[3,4])),
([1,2], [3,4])))
def testTupleSeqScalar(self):
self.assertTrue(recursive_equal(test_tuple_seq([1,2], 3), ([1,2],3)))
def testUnequalNested(self):
a = [[1, 2], [3, 4, 5]]
b = [[1, 2], [3, 4, 5, 6]]
self.assertFalse(recursive_equal(a, cuarray(b)))
def testUnequalContent(self):
a = [1, 2, 3]
b = [3, 2, 1]
self.assertFalse(recursive_equal(a, cuarray(b)))
def testUnequalLength(self):
a = [1, 2, 3]
b = [1, 2, 3, 4]
self.assertFalse(recursive_equal(a, cuarray(b)))
self.assertFalse(recursive_equal(b, cuarray(a)))
def testPythonNested(self):
a = [[[1, 2], [3, 4, 5]],
[[6, 7, 8, 9], [10, 11, 12, 13, 14], [15, 16, 17, 18, 19, 20]]]
self.assertTrue(recursive_equal(a, cuarray(a)))
def testPythonFlat(self):
a = [2.78, 3.14, 1.62]
self.assertTrue(recursive_equal(a, cuarray(a)))
def testNumpyFlat(self):
a = np.array([1, 2, 3, 4, 5])
self.assertTrue(recursive_equal(a, cuarray(a)))
def testReplNestedTuple(self):
a = test_repl(((1,2),3),2)
self.assertTrue(recursive_equal(
a,
[((1,2),3),((1,2),3)]))
def testReplInternalNamedTuple(self):
self.assertTrue(recursive_equal(test_internal_named_tuple_repl(1, 2),
[(1, 1), (1, 1)]))
def testTupleSeqSeq(self):
self.assertTrue(recursive_equal(test_tuple_seq([1,2], [3,4]), ([1,2],[3,4])))
def testZip(self):
self.assertTrue(
recursive_equal(test_zip(self.x, self.y),
[(1,3),(2,4),(3,5),(4,6),(5,7)]))
def testTupleSeqTuple(self):
self.assertTrue(recursive_equal(test_tuple_seq([1,2], (3,4)), ([1,2],(3,4))))
def testUnequalLength(self):
a = [1,2,3]
b = [1,2,3,4]
self.assertFalse(recursive_equal(a, cuarray(b)))
self.assertFalse(recursive_equal(b, cuarray(a)))
def testContainerize(self):
self.assertTrue(recursive_equal(test_containerize([1,2]), ([1,2], [-1,-2])))
def testUnequalNested(self):
a = [[1,2],[3,4,5]]
b = [[1,2],[3,4,5,6]]
self.assertFalse(recursive_equal(a, cuarray(b)))
def testPythonFlat(self):
a = [2.78, 3.14, 1.62]
self.assertTrue(recursive_equal(a, cuarray(a)))
def testSub(self):
source = [1,2,3,4,5]
result = test_sub(source)
self.assertTrue(recursive_equal([5,4,3,2,1],result))
def testPythonNested(self):
a = [[[1,2], [3,4,5]],
[[6,7,8,9], [10,11,12,13,14],
[15,16,17,18,19,20]]]
self.assertTrue(recursive_equal(a, cuarray(a)))
def testZip(self):
self.assertTrue(
recursive_equal(test_zip(self.x, self.y), [(1, 3), (2, 4), (3, 5),
(4, 6), (5, 7)]))
def testUnequalContent(self):
a = [1,2,3]
b = [3,2,1]
self.assertFalse(recursive_equal(a, cuarray(b)))
def testUnzip(self):
self.assertTrue(
recursive_equal(test_unzip(test_zip(self.x, self.y)),
([1,2,3,4,5],[3,4,5,6,7])))
def testAoS_2(self):
#XXX Change once cuarrays can be constructed with tuple elements
self.assertTrue(recursive_equal(self.golden_result_2,
test2(test(self.three))))
def testAoS_3(self):
self.assertTrue(recursive_equal(self.golden_result_2,
test3(self.three)))
def testReplTuple(self):
self.assertTrue(recursive_equal(test_repl((1, 1), 2), [(1, 1),
(1, 1)]))
def run_test(self, target, f, g, *args):
with target:
self.assertTrue(recursive_equal(f(*args), g))
def testReplInternalNamedTuple(self):
self.assertTrue(
recursive_equal(test_internal_named_tuple_repl(1, 2), [(1, 1),
(1, 1)]))
def testNestedTupleReturn(self):
self.assertTrue(
recursive_equal(test_nested_tuple_return(), (1, (2, 3))))
def run_test(self, target, f, g, *args):
with target:
self.assertTrue(recursive_equal(f(*args), g))
def testReplTuple(self):
self.assertTrue(recursive_equal(test_repl((1,1), 2), [(1,1),(1,1)]))
