def test_read_numpy_array_files_alpha_over_one(self): """ Checks the use of the alpha parameter in the loading of sift files. """ # Test loading with alpha = 0 point_array = sift_util.load_array_from_files( file_list=glob.glob('test_data/seminar.sift'), max_points=1369) self.assertEqual(point_array.shape[1], 128) point_array_1_0 = sift_util.load_array_from_files( file_list=glob.glob('test_data/seminar.sift'), alpha=1.0, max_points=1369) self.assertEqual(point_array_1_0.shape[1], 130) point_array_2_0 = sift_util.load_array_from_files( file_list=glob.glob('test_data/seminar.sift'), alpha=2.0, max_points=1369) self.assertEqual(point_array_2_0.shape[1], 130) for d_1, d_2 in itertools.izip(point_array_1_0, point_array_2_0): self.assertTrue( math.fabs(d_1[-2] * 2.0 - d_2[-2]) <= 1, 'd_1[-2] = %d, d_2[-2] = %d' % (d_1[-2], d_2[-2])) self.assertTrue( math.fabs(d_1[-1] * 2.0 - d_2[-1]) <= 1, 'd_1[-1] = %d, d_2[-1] = %d' % (d_1[-1], d_2[-1]))
def test_read_numpy_array_files_alpha_over_one(self): """ Checks the use of the alpha parameter in the loading of sift files. """ # Test loading with alpha = 0 point_array = sift_util.load_array_from_files(file_list=glob.glob('test_data/seminar.sift'), max_points=1369) self.assertEqual(point_array.shape[1], 128) point_array_1_0 = sift_util.load_array_from_files(file_list=glob.glob('test_data/seminar.sift'), alpha=1.0, max_points=1369) self.assertEqual(point_array_1_0.shape[1], 130) point_array_2_0 = sift_util.load_array_from_files(file_list=glob.glob('test_data/seminar.sift'), alpha=2.0, max_points=1369) self.assertEqual(point_array_2_0.shape[1], 130) for d_1, d_2 in itertools.izip(point_array_1_0, point_array_2_0): self.assertTrue(math.fabs(d_1[-2] * 2.0 - d_2[-2]) <= 1, 'd_1[-2] = %d, d_2[-2] = %d' % (d_1[-2], d_2[-2])) self.assertTrue(math.fabs(d_1[-1] * 2.0 - d_2[-1]) <= 1, 'd_1[-1] = %d, d_2[-1] = %d' % (d_1[-1], d_2[-1]))
def test_read_numpy_array_from_files_noalpha(self): """ Check the reading of sift descriptors from files. This doesn't check the application of the alpha parameter, just number of descriptors loaded and that max_points is being observed. """ # Test loading from empty list point_array = sift_util.load_array_from_files(file_list=[], max_points=300) self.assertEqual(len(point_array), 0) # Test loading with max points less than num available point_array = sift_util.load_array_from_files( file_list=glob.glob('test_data/seminar.sift'), max_points=300) self.assertTrue(len(point_array) > 0) self.assertTrue(len(point_array) <= 300) # Test loading with max points equal to num available point_array = sift_util.load_array_from_files( file_list=glob.glob('test_data/seminar.sift'), max_points=1369) self.assertEqual(len(point_array), 1369) # Test loading with max points not specified point_array = sift_util.load_array_from_files( file_list=glob.glob('test_data/seminar.sift')) self.assertEqual(len(point_array), 1369) # Test loading with max points higher than num available point_array = sift_util.load_array_from_files( file_list=glob.glob('test_data/seminar.sift'), max_points=1500) self.assertEqual(len(point_array), 1369) # Test loading from list with max points lower than available point_array = sift_util.load_array_from_files( file_list=glob.glob('test_data/*.sift'), max_points=1369) self.assertTrue(len(point_array) <= 1369) # Test loading from list with max points lower than available, but higher than a single file point_array = sift_util.load_array_from_files( file_list=glob.glob('test_data/*.sift'), max_points=1500) self.assertTrue( len(point_array) > 1369) # Check that more descriptors than from a single file self.assertTrue( len(point_array) <= 1500) # Check that within max_points
def test_read_numpy_array_from_files_noalpha(self): """ Check the reading of sift descriptors from files. This doesn't check the application of the alpha parameter, just number of descriptors loaded and that max_points is being observed. """ # Test loading from empty list point_array = sift_util.load_array_from_files(file_list=[], max_points=300) self.assertEqual(len(point_array), 0) # Test loading with max points less than num available point_array = sift_util.load_array_from_files( file_list=glob.glob('test_data/seminar.sift'), max_points=300) self.assertTrue(len(point_array) > 0) self.assertTrue(len(point_array) <= 300) # Test loading with max points equal to num available point_array = sift_util.load_array_from_files( file_list=glob.glob('test_data/seminar.sift'), max_points=1369) self.assertEqual(len(point_array), 1369) # Test loading with max points not specified point_array = sift_util.load_array_from_files( file_list=glob.glob('test_data/seminar.sift')) self.assertEqual(len(point_array), 1369) # Test loading with max points higher than num available point_array = sift_util.load_array_from_files( file_list=glob.glob('test_data/seminar.sift'), max_points=1500) self.assertEqual(len(point_array), 1369) # Test loading from list with max points lower than available point_array = sift_util.load_array_from_files( file_list=glob.glob('test_data/*.sift'), max_points=1369) self.assertTrue(len(point_array) <= 1369) # Test loading from list with max points lower than available, but higher than a single file point_array = sift_util.load_array_from_files( file_list=glob.glob('test_data/*.sift'), max_points=1500) self.assertTrue(len(point_array) > 1369) # Check that more descriptors than from a single file self.assertTrue(len(point_array) <= 1500) # Check that within max_points