class MorseTest(unittest.TestCase): def setUp(self): self.morse = Morse() def test_A(self): self.assertEqual(self.morse.encode('A'),['10111'])