python - 从哪里开始空课？

Question

这是关于 Python 的。我正在创建一个 BitCode 类，该字符串只能由 0 和 1 组成，如果不是，它会引发一个带有消息的 ValueError。

例如，

b = BitCode('1001')

有效。

写完后我在想:

class Bitcode:

代码的测试部分是:

class TestBitCode(unittest.TestCase):

    def test_constructor_value_error_bits(self):
        with self.assertRaises(ValueError):
            b = BitList('FE')

我在哪里可以放置检查以确保每个对象的参数只包含 0 和 1 的代码？

Answer 1

class BitCode:
    def __init__(self, s):
        if len(set(s + "01")) != 2:
            raise ValueError
        ...

s + "01" 将"0"和"1"添加到结果字符串中，结果我们得到一组字符，其中至少有"0"和"1"”。使用 set 我们只保留独特的元素。其中会有0和1，以及所有不为0和1的元素，我们求出长度，如果不等于2，则抛出异常