1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
|
#!/usr/bin/python
from lego import *
import unittest
def init_legos(): #why can't i do this in test_lego?
global bricklist
print "hi!"
brick1 = Lego("brick1", num_pegs=4, num_holes=2)
brick2 = Lego("brick2", num_pegs=2, num_holes=1)
brick3 = Lego("brick3", num_holes=3)
brick4 = Lego("brick4", num_pegs=4)
brick5 = Lego("brick5", num_pegs=0, num_holes=0)
#print brick1.interfaces[0].__class__.__name__
bricklist = [brick1, brick2, brick3, brick4, brick5]
init_legos()
class TestLegos(unittest.TestCase):
def test_equations(self):
pass
def test_dimensions(self):
#TODO: load up a lego with bad dimensions (make sure it complains)
pass
def test_generate_cad(self):
#test whether or not it can generate a basic OpenCASCADE model
pass
def test_lego(self):
brick1 = bricklist[0]
self.assertTrue(len(brick1.interfaces) == 6)
self.assertTrue(len(brick1.pegs()) == 4)
self.assertTrue(len(brick1.holes()) == 2)
def test_options(self):
expected = "paste good results here"
results = bricklist[0].interfaces[0].options(bricklist)
self.assertEqual(results, expected)
#print results
def test_options_peg_peg_hole_hole(self):
'''good ol' O(n^2) checking. if you're going to be a useless test, might as well do a good job at it'''
for brick in bricklist:
#self.assertFalse(brick.interfaces[0].__class__ == brick.interfaces[0].connected.__class__)
for interface in brick.interfaces:
for opt in interface.options(bricklist):
if1, if2 = opt.interface1, opt.interface2
self.assertFalse(if1.type == 'stud' and if1.connected.type == 'antistud')
self.assertFalse(if1.__class__ == Hole and if2.connected.__class__ == Hole)
def test_all_filled_up(self):
'''there should be no more options involving brick1's interface0'''
##TODO connect all bricks in bricklist here
self.assertTrue(len(bricklist[0].interfaces[0].options(bricklist)) == 0)
if __name__ == '__main__':
unittest.main()
|
