def test_002_getPending(self): import uuid REPO_FILE = '/tmp/testedRepoDir%s' % uuid.uuid4().hex os.mkdir(REPO_FILE) NUM_ELEMENTS = 10 try: # Allocate repository repo = ExperimentRepository(REPO_FILE, readOnly=False) self.assertEqual(repo.getPending(), []) elements = [] for elem in xrange(NUM_ELEMENTS): elements.append(repo.addExperiment(elem)) self.assertEqual(repo.getPending(), []) for i in xrange(NUM_ELEMENTS): repo.require(elements[i],i+1) self.assertEqual(sorted(repo.getPending()), sorted(zip(elements,range(1,NUM_ELEMENTS+1)))) # increasing and decreasing requirement should change result for i in xrange(NUM_ELEMENTS): repo.setRequiredCount(elements[i],NUM_ELEMENTS-i) self.assertEqual(set(repo.getPending()), set(zip(elements,range(NUM_ELEMENTS,0,-1)))) finally: # remove the directory, to keep things clean shutil.rmtree(REPO_FILE)
def test_002_getPending(self): import uuid REPO_FILE = '/tmp/testedRepoDir%s' % uuid.uuid4().hex os.mkdir(REPO_FILE) NUM_ELEMENTS = 10 try: # Allocate repository repo = ExperimentRepository(REPO_FILE, readOnly=False) self.assertEqual(repo.getPending(), []) elements = [] for elem in xrange(NUM_ELEMENTS): elements.append(repo.addExperiment(elem)) self.assertEqual(repo.getPending(), []) for i in xrange(NUM_ELEMENTS): repo.require(elements[i], i + 1) self.assertEqual(sorted(repo.getPending()), sorted(zip(elements, range(1, NUM_ELEMENTS + 1)))) # increasing and decreasing requirement should change result for i in xrange(NUM_ELEMENTS): repo.setRequiredCount(elements[i], NUM_ELEMENTS - i) self.assertEqual(set(repo.getPending()), set(zip(elements, range(NUM_ELEMENTS, 0, -1)))) finally: # remove the directory, to keep things clean shutil.rmtree(REPO_FILE)