def testEmptyIntersection(self):
values, ix1, ix2 = collections.intersect([1, 2, 3], [4, 5, 6])
self.assertEqual(len(values), 0)
self.assertEqual(len(ix1), 0)
self.assertEqual(len(ix2), 0)
values, ix1, ix2 = collections.intersect([], [])
self.assertEqual(len(values), 0)
self.assertEqual(len(ix1), 0)
self.assertEqual(len(ix2), 0)
def testPartialIntersectionSkipNones(self):
v1 = [None, 1, None, None, 2, None, 3, None, 4]
v2 = [1, None, 2, None, 3, 4]
values, ix1, ix2 = collections.intersect(v1, v2, True)
self.assertEqual(values, [1, 2, 3, 4])
self.assertEqual(ix1, [1, 4, 6, 8])
self.assertEqual(ix2, [0, 2, 4, 5])
values, ix2, ix1 = collections.intersect(v2, v1, True)
self.assertEqual(values, [1, 2, 3, 4])
self.assertEqual(ix1, [1, 4, 6, 8])
self.assertEqual(ix2, [0, 2, 4, 5])
def testPartialIntersection6(self):
v1 = [1, 1, 2, 2, 3, 3]
v2 = [1, 2, 3]
values, ix1, ix2 = collections.intersect(v1, v2)
self.assertEqual(values, [1, 2, 3])
self.assertEqual(ix1, [0, 2, 4])
self.assertEqual(ix2, [0, 1, 2])
values, ix2, ix1 = collections.intersect(v2, v1)
self.assertEqual(values, [1, 2, 3])
self.assertEqual(ix1, [0, 2, 4])
self.assertEqual(ix2, [0, 1, 2])
def testPartialIntersection2(self):
values, ix1, ix2 = collections.intersect([1, 2, 4], [1, 2, 3])
self.assertEqual(len(values), 2)
self.assertEqual(values[0], 1)
self.assertEqual(values[1], 2)
self.assertEqual(ix1[0], 0)
self.assertEqual(ix1[1], 1)
self.assertEqual(ix2[0], 0)
self.assertEqual(ix2[1], 1)
def testFullIntersectionWithDateTimes(self):
size = 1000
dateTimes1 = []
dateTimes2 = []
now = datetime.datetime.now()
for i in range(size):
dateTimes1.append(now + datetime.timedelta(seconds=i))
dateTimes2.append(now + datetime.timedelta(seconds=i))
self.assertEqual(dateTimes1, dateTimes2)
values, ix1, ix2 = collections.intersect(dateTimes1, dateTimes2)
self.assertEqual(values, dateTimes1)
self.assertEqual(values, dateTimes2)
self.assertEqual(ix1, list(range(size)))
self.assertEqual(ix1, ix2)
def testFullIntersection(self):
values, ix1, ix2 = collections.intersect([1, 2, 3], [1, 2, 3])
self.assertEqual(len(values), 3)
self.assertEqual(len(ix1), 3)
self.assertEqual(len(ix2), 3)
self.assertEqual(ix1, ix2)
def testPartialIntersection5(self):
values, ix1, ix2 = collections.intersect([1, 2, 3], [3, 6])
self.assertEqual(len(values), 1)
self.assertEqual(values[0], 3)
self.assertEqual(ix1[0], 2)
self.assertEqual(ix2[0], 0)