Beispiel #1
0
 def test_calculate_overlap(self):
     input_vector = np.array([[1, 0], [0, 0]])
     self.columns[0, 0, 0, 0] = 1
     min_overlap = 1
     connect_threshold = 0.1
     boost = np.array([[1, 1], [1, 1]])
     part_deque = partial(RingBuffer, input_array=np.zeros(2), copy=True)
     overlap_sum = defaultdict(part_deque)
     overlap, overlap_sum =\
         BSP.calculate_overlap(input_vector, self.columns, min_overlap,
                               connect_threshold, boost, overlap_sum)
     self.assertIsInstance(overlap, np.ndarray)
     self.assertEqual(overlap.shape, (2, 2))
     self.assertIsInstance(overlap_sum, defaultdict)
     self.assertListEqual(overlap.tolist(), [[1., 0.], [0., 0.]])
     self.assertIsInstance(overlap_sum[0, 0], RingBuffer)
     self.assertListEqual(list(overlap_sum[0, 0]), [0, 1])