def test_build_index_with_cache(self): # Empty memory data elements for storage empty_data = 'base64://' f = FlannNearestNeighborsIndex(empty_data, empty_data, empty_data) # Internal elements should initialize have zero-length byte values self.assertEqual(len(f._index_elem.get_bytes()), 0) self.assertEqual(len(f._index_param_elem.get_bytes()), 0) self.assertEqual(len(f._descr_cache_elem.get_bytes()), 0) # Make unit vectors, one for each feature dimension. dim = 8 test_descriptors = [] for i in range(dim): v = numpy.zeros(dim, float) v[i] = 1. d = DescriptorMemoryElement('unit', i) d.set_vector(v) test_descriptors.append(d) f.build_index(test_descriptors) # Internal elements should not have non-zero byte values. self.assertGreater(len(f._index_elem.get_bytes()), 0) self.assertGreater(len(f._index_param_elem.get_bytes()), 0) self.assertGreater(len(f._descr_cache_elem.get_bytes()), 0)
def test_build_index(self): # Empty memory data elements for storage empty_data = 'base64://' f = FlannNearestNeighborsIndex(empty_data, empty_data, empty_data) # Internal elements should initialize have zero-length byte values self.assertEqual(len(f._index_elem.get_bytes()), 0) self.assertEqual(len(f._index_param_elem.get_bytes()), 0) self.assertEqual(len(f._descr_cache_elem.get_bytes()), 0) # Make unit vectors, one for each feature dim = 8 test_descriptors = [] for i in range(dim): v = numpy.zeros(dim, float) v[i] = 1. d = DescriptorMemoryElement('unit', i) d.set_vector(v) test_descriptors.append(d) f.build_index(test_descriptors) # Internal elements should not have non-zero byte values. self.assertGreater(len(f._index_elem.get_bytes()), 0) self.assertGreater(len(f._index_param_elem.get_bytes()), 0) self.assertGreater(len(f._descr_cache_elem.get_bytes()), 0)