Testen Von Python-Skripten

Python-eigenen test-suite macht dies schon ein wenig, und wir nutzen im wesentlichen zwei Techniken:

1. Umleiten von stdout (wie andere schon vorgeschlagen). Wir verwenden ein context-manager:

   import io
   import sys
   import contextlib
   
   @contextlib.contextmanager
   def captured_output(stream_name):
       """Run the 'with' statement body using a StringIO object in place of a
          specific attribute on the sys module.
          Example use (with 'stream_name=stdout'):
   
          with captured_stdout() as s:
              print("hello")
              assert s.getvalue() == "hello"
       """
       orig_stdout = getattr(sys, stream_name)
       setattr(sys, stream_name, io.StringIO())
       try:
           yield getattr(sys, stream_name)
       finally:
           setattr(sys, stream_name, orig_stdout)
   
   def captured_stdout():
       return captured_output("stdout")
   
   def captured_stderr():
       return captured_output("stderr")
   
   def captured_stdin():
       return captured_output("stdin")

2. Mithilfe der subprocess Modul. Wir verwenden, wenn wir explizit testen wollen Umgang mit Kommandozeilen-Argumente. Sehen http://hg.python.org/cpython/file/default/Lib/test/test_cmd_line_script.py für mehrere Beispiele.

Hier ist etwas, das schrieb ich eines abends, dass tests Skript ausgeführt wird. Beachten Sie, dass der test deckt die grundlegenden Fälle, aber es ist nicht gründlich genug sein, um ein unittest von selbst. Halte es für einen ersten Entwurf.

import sys
import subprocess

if sys.platform == "win32":
   cmd = "zs.py"
else:
   cmd = "./zs.py"

def testrun(cmdline):
   try:
      retcode = subprocess.call(cmdline, shell=True)
      if retcode < 0:
         print >>sys.stderr, "Child was terminated by signal", -retcode
      else:
         return retcode
   except OSError, e:
      return e

tests = []
tests.append( (0, " string pattern 4") )
tests.append( (1, " string pattern") )
tests.append( (3, " string pattern notanumber") )
passed = 0

for t in tests:
   r = testrun(cmd + t[1])
   if r == t[0]:
      res = "passed"
      passed += 1
   else:
      res = "FAILED"
   print res, r, t[1]

print
if passed != len(tests):
   print "only",passed,"tests passed"
else:
   print "all tests passed"

Und hier ist das Skript, das getestet wurde, zs.py Dies gilt Muster sucht in einem string ähnlich dem Weg, Biochemiker suchen von mustern in DNA-Daten oder protein-chain-Daten.

#!/usr/bin/env python

# zs - some example Python code to demonstrate to Z??s
#      interviewers that the writer really does know Python

import sys
from itertools import *

usage = '''
   Usage: zs <string> <pattern> <n>"
          print top n matches of pattern in substring"
'''

if sys.hexversion > 0x03000000:
   print "This script is only intended to run on Python version 2"
   sys.exit(2)

if len(sys.argv) != 4:
   print usage
   sys.exit(1)

A = sys.argv[1] # string to be searched
B = sys.argv[2] # pattern being searched for
N = sys.argv[3] # number of matches to report

if not N.isdigit():
   print "<n> must be a number"
   print usage
   sys.exit(3)

def matchscore(s1, s2):
   ''' a helper function to calculate the match score
   '''
   matches = 0
   for i in xrange(len(s1)):
      if s1[i] == s2[i]:
         matches += 1
   return (matches + 0.0) / len(s1)  # added 0.0 to force floating point div

def slices(s, n):
   ''' this is a generator that returns the sequence of slices of
       the input string s that are n characters long '''
   slen = len(s)
   for i in xrange(slen - n + 1):
      yield s[i:i+n]

matchlen = len(B)
allscores = ((matchscore(x,B),x,i) for i,x in enumerate(slices(A,matchlen)))
nonzeros = [ y for y in allscores if y[0] != 0 ]

for elem in sorted(nonzeros,key=lambda e: e[0],reverse=True):
   nprinted = 0 # We will count them; in case num elements > N
   print elem[1], str(round(elem[0],4)), elem[2]
   nprinted += 1
   if nprinted >= N:
      break