コード例 #1
0
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()
コード例 #2
0
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)