mirror of
https://github.com/coding-horror/basic-computer-games.git
synced 2025-12-22 07:10:42 -08:00
8495e59a8f
Avoid executing code on module level as this prevents importing the module for testing. Especially infinite loops are evil.
50 lines
1.0 KiB
Python
50 lines
1.0 KiB
Python
import unittest
|
|
|
|
import tower
|
|
|
|
|
|
class TowerTestCase(unittest.TestCase):
|
|
def test_something(self):
|
|
t = tower.Tower()
|
|
self.assertTrue(t.empty())
|
|
|
|
d = tower.Disk(3)
|
|
t.add(d)
|
|
self.assertFalse(t.empty())
|
|
|
|
d5 = tower.Disk(5)
|
|
self.assertRaises(Exception, t.add, d5)
|
|
self.assertFalse(t.empty())
|
|
|
|
def test_oksize(self):
|
|
t = tower.Tower()
|
|
self.assertTrue(t.empty())
|
|
|
|
d5 = tower.Disk(5)
|
|
t.add(d5)
|
|
self.assertFalse(t.empty())
|
|
|
|
d3 = tower.Disk(3)
|
|
t.add(d3)
|
|
self.assertFalse(t.empty())
|
|
|
|
self.assertEqual(t.top(), d3)
|
|
self.assertEqual(t.pop(), d3)
|
|
self.assertEqual(t.pop(), d5)
|
|
|
|
def test_game(self):
|
|
g = tower.Game()
|
|
self.assertEqual(g.moves(), 0)
|
|
self.assertFalse(g.winner())
|
|
|
|
def test_format(self):
|
|
t = tower.Tower()
|
|
d3 = tower.Disk(3)
|
|
d5 = tower.Disk(5)
|
|
t.add(d5)
|
|
t.add(d3)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|