12
12
13
13
class TestImageMultiDiscrete (unittest .TestCase ):
14
14
def test_image_multi_discrete (self ):
15
+ render = False
16
+
15
17
ds4 = Discrete (4 )
16
18
ds4 = [ds4 .n ]
17
19
print (ds4 )
@@ -20,7 +22,8 @@ def test_image_multi_discrete(self):
20
22
21
23
# img1 = Image.fromarray(imd.disjoint_states[0][1], 'L')
22
24
img1 = Image .fromarray (np .squeeze (imd .get_concatenated_image (3 )), "L" )
23
- img1 .show ()
25
+ if render :
26
+ img1 .show ()
24
27
25
28
imd = ImageMultiDiscrete (
26
29
ds4 ,
@@ -29,7 +32,8 @@ def test_image_multi_discrete(self):
29
32
cust_path = "/home/rajanr/textures" ,
30
33
)
31
34
img1 = Image .fromarray (np .squeeze (imd .get_concatenated_image (2 )), "RGB" )
32
- img1 .show ()
35
+ if render :
36
+ img1 .show ()
33
37
34
38
imd = ImageMultiDiscrete (
35
39
ds4 ,
@@ -38,58 +42,74 @@ def test_image_multi_discrete(self):
38
42
cust_path = "/home/rajanr/textures" ,
39
43
)
40
44
img1 = Image .fromarray (np .squeeze (imd .get_concatenated_image (1 )), "RGB" )
41
- img1 .show ()
45
+ if render :
46
+ img1 .show ()
42
47
43
48
imd = ImageMultiDiscrete (ds4 , transforms = "scale" )
44
49
img1 = Image .fromarray (np .squeeze (imd .get_concatenated_image (3 )), "L" )
45
- img1 .show ()
50
+ if render :
51
+ img1 .show ()
46
52
47
53
imd = ImageMultiDiscrete (ds4 , transforms = "rotate" )
48
54
img1 = Image .fromarray (np .squeeze (imd .get_concatenated_image (3 )), "L" )
49
- img1 .show ()
55
+ if render :
56
+ img1 .show ()
50
57
51
58
imd = ImageMultiDiscrete (ds4 , transforms = "flip" )
52
59
img1 = Image .fromarray (np .squeeze (imd .get_concatenated_image (3 )), "L" )
53
- img1 .show ()
60
+ if render :
61
+ img1 .show ()
54
62
55
63
imd = ImageMultiDiscrete (ds4 , transforms = "shift" )
56
64
img1 = Image .fromarray (np .squeeze (imd .get_concatenated_image (2 )), "L" )
57
- img1 .show ()
65
+ if render :
66
+ img1 .show ()
58
67
imd = ImageMultiDiscrete (ds4 , transforms = "scale" )
59
68
img1 = Image .fromarray (np .squeeze (imd .get_concatenated_image (2 )), "L" )
60
- img1 .show ()
69
+ if render :
70
+ img1 .show ()
61
71
imd = ImageMultiDiscrete (ds4 , transforms = "rotate" )
62
72
img1 = Image .fromarray (np .squeeze (imd .get_concatenated_image (2 )), "L" )
63
- img1 .show ()
73
+ if render :
74
+ img1 .show ()
64
75
imd = ImageMultiDiscrete (ds4 , transforms = "flip" )
65
76
img1 = Image .fromarray (np .squeeze (imd .get_concatenated_image (2 )), "L" )
66
- img1 .show ()
77
+ if render :
78
+ img1 .show ()
67
79
68
80
imd = ImageMultiDiscrete (ds4 , transforms = "shift" )
69
81
img1 = Image .fromarray (np .squeeze (imd .get_concatenated_image (1 )), "L" )
70
- img1 .show ()
82
+ if render :
83
+ img1 .show ()
71
84
imd = ImageMultiDiscrete (ds4 , transforms = "scale" )
72
85
img1 = Image .fromarray (np .squeeze (imd .get_concatenated_image (1 )), "L" )
73
- img1 .show ()
86
+ if render :
87
+ img1 .show ()
74
88
imd = ImageMultiDiscrete (ds4 , transforms = "rotate" )
75
89
img1 = Image .fromarray (np .squeeze (imd .get_concatenated_image (1 )), "L" )
76
- img1 .show ()
90
+ if render :
91
+ img1 .show ()
77
92
imd = ImageMultiDiscrete (ds4 , transforms = "flip" )
78
93
img1 = Image .fromarray (np .squeeze (imd .get_concatenated_image (1 )), "L" )
79
- img1 .show ()
94
+ if render :
95
+ img1 .show ()
80
96
81
97
imd = ImageMultiDiscrete (ds4 , transforms = "shift" )
82
98
img1 = Image .fromarray (np .squeeze (imd .get_concatenated_image (0 )), "L" )
83
- img1 .show ()
99
+ if render :
100
+ img1 .show ()
84
101
imd = ImageMultiDiscrete (ds4 , transforms = "scale" )
85
102
img1 = Image .fromarray (np .squeeze (imd .get_concatenated_image (0 )), "L" )
86
- img1 .show ()
103
+ if render :
104
+ img1 .show ()
87
105
imd = ImageMultiDiscrete (ds4 , transforms = "rotate" )
88
106
img1 = Image .fromarray (np .squeeze (imd .get_concatenated_image (0 )), "L" )
89
- img1 .show ()
107
+ if render :
108
+ img1 .show ()
90
109
imd = ImageMultiDiscrete (ds4 , transforms = "flip" )
91
110
img1 = Image .fromarray (np .squeeze (imd .get_concatenated_image (0 )), "L" )
92
- img1 .show ()
111
+ if render :
112
+ img1 .show ()
93
113
94
114
# imd = ImageMultiDiscrete(ds4, 100, 100)
95
115
# from PIL import Image
0 commit comments