blob: 45b790201a01db362ed1358fc59ed65311269324 (
plain)
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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
# Copyright 2006-2007 Nanorex, Inc. See LICENSE file for details.
# $Id$
import iguana
import unittest
baseline_code = """
+!:
<atomic
/* x p */
dup fetch /* x p (*p) */
rot + swap /* (*p)+x p */
store
atomic> exit
foo:
rand exit
bar:
2.718 exit
quux:
ouch exit
/* this requires we set up some data memory */
tryloop:
0 0 store
600 0 do
i 0 +!
loop
exit
tryloop2:
0
600 0 do
i +
loop
exit
quickloop:
600 0 do loop exit
"""
class IguanaTests(unittest.TestCase):
def setUp(self):
self.prog = prog = iguana.Program()
prog.compile(baseline_code)
def test_loop1(self):
self.prog.compile("main: tryloop exit")
mem = 3 * [ 0.0 ]
self.prog.thread("main", mem)
self.prog.run()
assert mem[0] == (600.0 * 599) / 2
assert mem[1] == 0.0
assert mem[2] == 0.0
# this is quite slow, because we're doing lots of "fetch" and "store" operations
def test_loop1_many_threads(self):
self.prog.compile("main: tryloop exit")
mem = 3 * [ 0.0 ]
for i in range(1000):
self.prog.thread("main", mem)
self.prog.run()
assert mem[0] == 1000 * (600.0 * 599) / 2
assert mem[1] == 0.0
assert mem[2] == 0.0
def test_loop2(self):
self.prog.compile("main: tryloop2 exit")
T = self.prog.thread("main")
self.prog.run()
assert T.pop() == (600.0 * 599) / 2
def test_loop2_many_threads(self):
self.prog.compile("main: tryloop2 exit")
threads = [ ]
for i in range(1000):
threads.append(self.prog.thread("main"))
self.prog.run()
for i in range(1000):
assert threads[i].pop() == (600.0 * 599) / 2
def test_quickloop(self):
self.prog.compile("main: quickloop exit")
T = self.prog.thread("main")
self.prog.run()
def test_quux(self):
T = self.prog.thread("quux")
try:
self.prog.run()
assert False, "this should have failed"
except iguana.IguanaError:
pass
def test_bar(self):
T = self.prog.thread("bar")
self.prog.run()
assert T.pop() == 2.718
def __test_foo(self):
for i in range(20):
T = self.prog.thread("foo")
self.prog.run()
print T.pop()
def test():
suite = unittest.makeSuite(IguanaTests, 'test')
#suite = unittest.makeSuite(Tests, 'test_atomset_atomInfo')
runner = unittest.TextTestRunner()
runner.run(suite)
if __name__ == "__main__":
test()
|
