from preenlistment_module import PreenlistmentModule as PM
import sys
def main():
filename = ''
num_runs = 0
try:
filename = sys.argv[1]
with open(filename,'r') as f:
pass
except IndexError:
print 'No input! Please pass the .csv file as an argument.'
return
except IOError:
print 'Cannot read input file.'
return
try:
num_runs = int(sys.argv[2])
except Exception, e:
num_runs = 1
pm = PM()
pm.load(filename)
pm.batchrun(num_runs)
if __name__ == '__main__':
main()
def main():
# test if course gets initialized properly
# no conflict sections
course1 = Course('Subject 10', 'ABC', 10, 10, 3.0)
# 1 conflict section
course2 = Course('Subject 11', 'DEF', 10, 10, 3.0, 'GHI')
# multiple conflict sections
course3 = Course('Subject 12', 'JKL', 10, 10, 3.0, 'MNO', 'PQR', 'STU')
# initializing Course with whitespaces in args
course4 = Course(' Subject 13 ', ' ABC ', 10, 10, 3.0, ' LOL ')
print '\nSubject w/o conflict sections'
print course1
print '\nSubject w/ 1 conflict section'
print course2
print '\nSubject w/ multiple conflict sections'
print course3
print '\ninitializing Course with whitespaces in args'
print course4
# test if class is added to a desired classes succesfully
print '\ntest if class is added to a desired classes succesfully'
pm = PM()
pm.add_class(*['Subject 13','VWX', 10, 10, 3.0, 'YZA', 'BCD', 'EFG'])
print pm
# test if preenlistment module loads .csv file successfully
print '\ntest if preenlistment module loads .csv file successfully'
pm.load('sample.csv')
print pm
# test if the conflict resolution works
print '\ntest if the conflict resolution works'
print 'conflict between...'
print pm.desired_classes[0], pm.desired_classes[1]
print pm.has_conflict(pm.desired_classes[0],pm.desired_classes[1])
print pm.desired_classes[0], pm.desired_classes[2]
print pm.has_conflict(pm.desired_classes[0],pm.desired_classes[2])
print pm.desired_classes[0], pm.desired_classes[3]
print pm.has_conflict(pm.desired_classes[0],pm.desired_classes[3])
print '\n\nSimulating a batchrun (5 runs)'
pm.load('save.csv')
pm.batchrun(5)
