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)
