7
7
class TestHsi2Rgb (unittest .TestCase ):
8
8
def setUp (self ):
9
9
return super ().setUp ()
10
- Path ("tests/runs/" ).mkdir (parents = True , exist_ok = True )
11
10
12
11
def test_batch_tensor (self ):
13
12
import torch
@@ -49,6 +48,7 @@ def test_muufl(self):
49
48
self .assertEqual (rgb .shape , (hsi .shape [1 ], hsi .shape [2 ], 3 ))
50
49
self .assertGreaterEqual (rgb .max (), 0 )
51
50
self .assertLessEqual (rgb .min (), 1 )
51
+ Path ("tests/runs/" ).mkdir (parents = True , exist_ok = True )
52
52
imsave ('tests/runs/test_muufl.png' , rgb )
53
53
54
54
def test_trento (self ):
@@ -58,6 +58,7 @@ def test_trento(self):
58
58
self .assertEqual (rgb .shape , (hsi .shape [1 ], hsi .shape [2 ], 3 ))
59
59
self .assertGreaterEqual (rgb .max (), 0 )
60
60
self .assertLessEqual (rgb .min (), 1 )
61
+ Path ("tests/runs/" ).mkdir (parents = True , exist_ok = True )
61
62
imsave ('tests/runs/test_trento.png' , rgb )
62
63
63
64
def test_houston2013 (self ):
@@ -67,6 +68,7 @@ def test_houston2013(self):
67
68
self .assertEqual (rgb .shape , (hsi .shape [1 ], hsi .shape [2 ], 3 ))
68
69
self .assertGreaterEqual (rgb .max (), 0 )
69
70
self .assertLessEqual (rgb .min (), 1 )
71
+ Path ("tests/runs/" ).mkdir (parents = True , exist_ok = True )
70
72
imsave ('tests/runs/test_houston2013.png' , rgb )
71
73
72
74
def test_jax_tensor (self ):
0 commit comments