mirror of
https://github.com/donnemartin/interactive-coding-challenges.git
synced 2024-03-22 13:11:13 +08:00
29 lines
863 B
Python
29 lines
863 B
Python
import unittest
|
|
|
|
|
|
class TestBitsScreen(unittest.TestCase):
|
|
|
|
def test_draw_line(self):
|
|
bits_screen = BitsScreen()
|
|
screen = []
|
|
for _ in range(20):
|
|
screen.append(int('00000000', base=2))
|
|
bits_screen.draw_line(screen, width=32, x1=68, x2=80)
|
|
self.assertEqual(screen[8], int('00001111', base=2))
|
|
self.assertEqual(screen[9], int('11111111', base=2))
|
|
self.assertEqual(screen[10], int('10000000', base=2))
|
|
bits_screen.draw_line(screen, width=32, x1=2, x2=6)
|
|
self.assertEqual(screen[0], int('00111110', base=2))
|
|
bits_screen.draw_line(screen, width=32, x1=10, x2=13)
|
|
self.assertEqual(screen[1], int('00111100', base=2))
|
|
print('Success: test_draw_line')
|
|
|
|
|
|
def main():
|
|
test = TestBitsScreen()
|
|
test.test_draw_line()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|