#!/usr/bin/python # # Tests for comparenum. # Does not require eshoutput.py # # import sys, imp, atexit sys.path.append("/home/courses/cs3214/software/pexpect-dpty/"); import pexpect, shellio, signal, time, os, re, proc_check #Ensure the shell process is terminated def force_shell_termination(shell_process): c.close(force=True) #pulling in the regular expression and other definitions definitions_scriptname = sys.argv[1] plugin_dir = sys.argv[2] def_module = imp.load_source('', definitions_scriptname) logfile = None if hasattr(def_module, 'logfile'): logfile = def_module.logfile #spawn an instance of the shell c = pexpect.spawn(def_module.shell + plugin_dir, drainpty=True, logfile=logfile) atexit.register(force_shell_termination, shell_process=c) c.sendline("comparenum 1 5") assert c.expect("1.000 is less than 5.000") == 0, ""; c.sendline("comparenum 4.2 1.3") assert c.expect("4.200 is bigger than 1.300") == 0, ""; c.sendline("comparenum 3 3") assert c.expect("They're the same!") == 0, ""; shellio.success()