Вызов командной строки unittests из __main__ failing

Question

Я решаю некоторые упражнения на Python и используя unittest для автоматизации некоторых проверок моего кода. Одна программа запускает единственный unittest просто отлично, и он проходит. Вторая дает следующее сообщение об ошибке:

$ python s1c6.py 
E 
====================================================================== 
ERROR: s1c6 (unittest.loader._FailedTest) 
---------------------------------------------------------------------- 
AttributeError: module '__main__' has no attribute 's1c6' 

---------------------------------------------------------------------- 
Ran 1 test in 0.001s 

FAILED (errors=1) 

Вот код для рабочего сценария:

# s1c5.py 
import unittest 

import cryptopals 


class TestRepeatingKeyXor(unittest.TestCase): 
    def testCase(self): 
     key = b"ICE" 
     data = b"Burning 'em, if you ain't quick and nimble\nI go crazy when I hear a cymbal" 
     expected = bytes.fromhex(
      "0b3637272a2b2e63622c2e69692a23693a2a3c6324202d623d63343c2a26226324272765272a282b2f20430a652e2c652a3124333a653e2b2027630c692b20283165286326302e27282f") 
     self.assertEqual(expected, cryptopals.xorcrypt(key, data)) 


if __name__ == "__main__": 
    unittest.main() 

И код для неисправного сценария:

# s1c6.py 
import unittest 
import bitstring 

import cryptopals 


class TestHammingDistance(unittest.TestCase): 
    def testCase(self): 
     str1 = b'this is a test' 
     str2 = b'wokka wokka!!!' 
     expected = 37 
     self.assertEqual(expected, hamming_distance(str1, str2)) 


def hamming_distance(str1, str2): 
    temp = cryptopals.xor(str1, str2) 
    return sum(bitstring.Bits(temp)) 


if __name__ == "__main__": 
    unittest.main() 

Я не вижу фундаментальный разница между этими двумя программами, которые могут вызвать ошибку в одном, а не в другом. Что мне не хватает?

import itertools 
import operator 


def xor(a, b): 
    return bytes(map(operator.xor, a, b)) 


def xorcrypt(key, cipher): 
    return b''.join(xor(key, x) for x in grouper(cipher, len(key))) 


def grouper(iterable, n): 
    it = iter(iterable) 
    group = tuple(itertools.islice(it, n)) 
    while group: 
     yield group 
     group = tuple(itertools.islice(it, n)) 

версия "Raw" из неисправного сценария:

# s1c6_raw.py 
import cryptopals 

key = b"ICE" 
data = b"Burning 'em, if you ain't quick and nimble\nI go crazy when I hear a cymbal" 
expected = bytes.fromhex(
    "0b3637272a2b2e63622c2e69692a23693a2a3c6324202d623d63343c2a26226324272765272a282b2f20430a652e2c652a3124333a653e2b2027630c692b20283165286326302e27282f") 
print(cryptopals.xorcrypt(key, data)) 

Вышеприведенные работает отлично и печатает ожидаемый выход.

Answer 1

Проблема заключается в том, что я бегу два сценария по-разному:

$ python s1c5.py 

и

$ python s1c6.py s1c6.txt 

Поскольку unittest.main() разбирает аргументы командной строки, есть ошибка во втором случае. Если я передаю аргумент командной строки первой программе, я также получаю ту же ошибку.