isl/dist/isl_test_python.py

8 import isl
10 # Test that isl objects can be constructed.
24 	zero1 = isl.val("0")
27 	zero2 = isl.val(0)
30 	zero3 = isl.val.zero()
33 	bs = isl.basic_set("{ [1] }")
34 	result = isl.set("{ [1] }")
35 	s = isl.set(bs)
38 	us = isl.union_set("{ A[1]; B[2, 3] }")
39 	empty = isl.union_set.empty()
45 	val_int = isl.val(i)
46 	val_str = isl.val(str(i))
58 # Test isl objects parameters.
60 # Verify that isl objects can be passed as lvalue and rvalue parameters.
61 # Also verify that isl object parameters are automatically type converted if
63 # any additional parameters, apart from the isl object on which
67 	a = isl.set("{ [0] }")
68 	b = isl.set("{ [1] }")
69 	c = isl.set("{ [2] }")
70 	expected = isl.set("{ [i] : 0 <= i <= 2 }")
79 	a2 = isl.basic_set("{ [0] }")
82 	two = isl.val(2)
83 	half = isl.val("1/2")
89 # This includes integer and isl object parameters.
95 # Test that isl objects are returned correctly.
101 	one = isl.val("1")
102 	two = isl.val("2")
103 	three = isl.val("3")
112 	one = isl.val("1")
113 	neg_one = isl.val("-1")
114 	zero = isl.val("0")
125 	empty = isl.set("{ : false }")
126 	univ = isl.set("{ : }")
135 # Do so by calling overloaded isl.ast_build.from_expr methods.
138 	context = isl.set("[n] -> { : }")
139 	build = isl.ast_build.from_context(context)
140 	pw_aff = isl.pw_aff("[n] -> { [n] }")
141 	set = isl.set("[n] -> { : n >= 0 }")
153 # Test that isl objects, integers, boolean values, and strings are
162 # A class that is used to test isl.id.user.
168 # Test isl.id.user.
174 	id = isl.id("test", 5)
175 	id2 = isl.id("test2")
176 	id3 = isl.id("S", S())
191 	s = isl.set("{ [0]; [1]; [2] }")
222 	list = isl.id_list(3)
223 	sorted = [isl.id_list(3)]
225 		'a' : isl.map("{ [0] -> [1] }"),
226 		'b' : isl.map("{ [1] -> [0] }"),
227 		'c' : isl.map("{ [i = 0:1] -> [i] }"),
252 	us = isl.union_set("{ A[i]; B[j] }")
263 		return s.is_subset(isl.set("{ A[x] }"))
267 		return not s.is_subset(isl.set("{ A[x] }"))
283 	unit = isl.space.unit()
287 	set = isl.set.universe(set_space)
288 	map = isl.map.universe(map_space)
289 	assert(set.is_equal(isl.set("{ A[*,*,*] }")))
290 	assert(map.is_equal(isl.map("{ A[*,*,*] -> B[*,*] }")))
297 	A = isl.union_set("{ A[i] : 0 <= i < 10 }")
298 	B = isl.union_set("{ B[i] : 0 <= i < 20 }")
300 	node = isl.schedule_node.from_domain(A.union(B))
303 	filters = isl.union_set_list(A).add(B)
306 	f_A = isl.multi_union_pw_aff("[ { A[i] -> [i] } ]")
313 	f_B = isl.multi_union_pw_aff("[ { B[i] -> [i] } ]")
333 	assert(type(root) == isl.schedule_node_domain)
367 		return type(node) != isl.schedule_node_domain
381 	filters = [isl.union_set("{}")]
383 		if type(node) == isl.schedule_node_filter:
398 		if type(node) == isl.schedule_node_band:
409 	build = isl.ast_build()
429 	build = isl.ast_build()
448 	build = isl.ast_build()
472 	pa = isl.pw_aff("[n] -> { [n + 1] }")
473 	build = isl.ast_build.from_context(pa.domain())
476 	assert(type(op) == isl.ast_expr_op_add)
479 # Test the isl Python interface
485 #  - isl.id.user