Advertisement
Guest User

check.py

a guest
Aug 14th, 2014
294
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.64 KB | None | 0 0
  1. import sys
  2. import subprocess as sb
  3. from hashlib import md5
  4.  
  5. cases = (
  6. """
  7. ATG      2e4a906c44a96fe84134bf4346adf11c
  8. C        e3648b8960967463784818c3eee57246
  9. TTT      6028a90b05775905ef1a00e7a45463c5
  10. TAGCAT   3b834d2b7b9adc4113ffabd52d354c41
  11. GATTACA  a19463f965c641d071e07da59d64a418
  12. """).strip().split('\n')
  13.  
  14. for input, checksum in map(str.split, cases):
  15.     proc = sb.Popen(sys.argv[1:], stdin=sb.PIPE,stdout=sb.PIPE)
  16.     output = proc.communicate(input.encode())[0]
  17.     output = b'\n'.join(map(bytes.rstrip, output.rstrip().split(b'\n')))
  18.     checksum_output = md5(output).hexdigest()
  19.     assert checksum_output == checksum, input
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement