class TestSevenSegmentsDisplay(unittest.TestCase): def setUp(self): self.ssd = SevenSegmentsDisplay(1, [(' ', '-', ' '), ('|', ' ', '|'), (' ', '-', ' '), ('|', ' ', '|'), (' ', '-', ' ')]) def test_topLeft(self): self.assertEqual(' ', self.ssd.topLeft()) def test_topCenter(self): self.assertEqual('-', self.ssd.topCenter()) def test_topRight(self): self.assertEqual(' ', self.ssd.topRight()) def test_midTopLeft(self): self.assertEqual('|', self.ssd.midTopLeft()) def test_midTopCenter(self): self.assertEqual(' ', self.ssd.midTopCenter()) def test_midTopRight(self): self.assertEqual('|', self.ssd.midTopRight()) def test_midLeft(self): self.assertEqual(' ', self.ssd.midLeft()) def test_midCenter(self): self.assertEqual('-', self.ssd.midCenter()) def test_midRight(self): self.assertEqual(' ', self.ssd.midRight()) def test_midBottonLeft(self): self.assertEqual('|', self.ssd.midBottonLeft()) def test_midBottonCenter(self): self.assertEqual(' ', self.ssd.midBottonCenter()) def test_midBottonRight(self): self.assertEqual('|', self.ssd.midBottonRight()) def test_bottonLeft(self): self.assertEqual(' ', self.ssd.bottonLeft()) def test_bottonCenter(self): self.assertEqual('-', self.ssd.bottonCenter()) def test_bottonRight(self): self.assertEqual(' ', self.ssd.bottonRight())
class TestSevenSegmentsDisplayNumbers(unittest.TestCase): def test_A(self): self.ssd = SevenSegmentsDisplay(1, [(' ', '-', ' '), ('|', ' ', '|'), (' ', '-', ' '), ('|', ' ', '|'), (' ', ' ', ' ')]) self.assertEqual(' ', self.ssd.topLeft()) self.assertEqual('-', self.ssd.topCenter()) self.assertEqual(' ', self.ssd.topRight()) self.assertEqual('|', self.ssd.midTopLeft()) self.assertEqual(' ', self.ssd.midTopCenter()) self.assertEqual('|', self.ssd.midTopRight()) self.assertEqual(' ', self.ssd.midLeft()) self.assertEqual('-', self.ssd.midCenter()) self.assertEqual(' ', self.ssd.midRight()) self.assertEqual('|', self.ssd.midBottonLeft()) self.assertEqual(' ', self.ssd.midBottonCenter()) self.assertEqual('|', self.ssd.midBottonRight()) self.assertEqual(' ', self.ssd.bottonLeft()) self.assertEqual(' ', self.ssd.bottonCenter()) self.assertEqual(' ', self.ssd.bottonRight()) def test_B(self): self.ssd = SevenSegmentsDisplay(1, [(' ', '-', ' '), ('|', ' ', '|'), ('|', '-', ' '), ('|', ' ', '|'), (' ', '-', ' ')]) self.assertEqual(' ', self.ssd.topLeft()) self.assertEqual('-', self.ssd.topCenter()) self.assertEqual(' ', self.ssd.topRight()) self.assertEqual('|', self.ssd.midTopLeft()) self.assertEqual(' ', self.ssd.midTopCenter()) self.assertEqual('|', self.ssd.midTopRight()) self.assertEqual('|', self.ssd.midLeft()) self.assertEqual('-', self.ssd.midCenter()) self.assertEqual(' ', self.ssd.midRight()) self.assertEqual('|', self.ssd.midBottonLeft()) self.assertEqual(' ', self.ssd.midBottonCenter()) self.assertEqual('|', self.ssd.midBottonRight()) self.assertEqual(' ', self.ssd.bottonLeft()) self.assertEqual('-', self.ssd.bottonCenter()) self.assertEqual(' ', self.ssd.bottonRight()) def test_zero(self): self.ssd = SevenSegmentsDisplay(1, [(' ', '-', ' '), ('|', ' ', '|'), (' ', ' ', ' '), ('|', ' ', '|'), (' ', '-', ' ')]) self.assertEqual(' ', self.ssd.topLeft()) self.assertEqual('-', self.ssd.topCenter()) self.assertEqual(' ', self.ssd.topRight()) self.assertEqual('|', self.ssd.midTopLeft()) self.assertEqual(' ', self.ssd.midTopCenter()) self.assertEqual('|', self.ssd.midTopRight()) self.assertEqual(' ', self.ssd.midLeft()) self.assertEqual(' ', self.ssd.midCenter()) self.assertEqual(' ', self.ssd.midRight()) self.assertEqual('|', self.ssd.midBottonLeft()) self.assertEqual(' ', self.ssd.midBottonCenter()) self.assertEqual('|', self.ssd.midBottonRight()) self.assertEqual(' ', self.ssd.bottonLeft()) self.assertEqual('-', self.ssd.bottonCenter()) self.assertEqual(' ', self.ssd.bottonRight()) def test_one(self): self.ssd = SevenSegmentsDisplay(1, [(' ', ' ', ' '), (' ', ' ', '|'), (' ', ' ', ' '), (' ', ' ', '|'), (' ', ' ', ' ')]) self.assertEqual(' ', self.ssd.topLeft()) self.assertEqual(' ', self.ssd.topCenter()) self.assertEqual(' ', self.ssd.topRight()) self.assertEqual(' ', self.ssd.midTopLeft()) self.assertEqual(' ', self.ssd.midTopCenter()) self.assertEqual('|', self.ssd.midTopRight()) self.assertEqual(' ', self.ssd.midLeft()) self.assertEqual(' ', self.ssd.midCenter()) self.assertEqual(' ', self.ssd.midRight()) self.assertEqual(' ', self.ssd.midBottonLeft()) self.assertEqual(' ', self.ssd.midBottonCenter()) self.assertEqual('|', self.ssd.midBottonRight()) self.assertEqual(' ', self.ssd.bottonLeft()) self.assertEqual(' ', self.ssd.bottonCenter()) self.assertEqual(' ', self.ssd.bottonRight()) def test_two(self): self.ssd = SevenSegmentsDisplay(1, [(' ', '-', ' '), (' ', ' ', '|'), (' ', '-', ' '), ('|', ' ', ' '), (' ', '-', ' ')]) self.assertEqual(' ', self.ssd.topLeft()) self.assertEqual('-', self.ssd.topCenter()) self.assertEqual(' ', self.ssd.topRight()) self.assertEqual(' ', self.ssd.midTopLeft()) self.assertEqual(' ', self.ssd.midTopCenter()) self.assertEqual('|', self.ssd.midTopRight()) self.assertEqual(' ', self.ssd.midLeft()) self.assertEqual('-', self.ssd.midCenter()) self.assertEqual(' ', self.ssd.midRight()) self.assertEqual('|', self.ssd.midBottonLeft()) self.assertEqual(' ', self.ssd.midBottonCenter()) self.assertEqual(' ', self.ssd.midBottonRight()) self.assertEqual(' ', self.ssd.bottonLeft()) self.assertEqual('-', self.ssd.bottonCenter()) self.assertEqual(' ', self.ssd.bottonRight())