9
9
10
10
class TestClass :
11
11
12
- parameters = {}
12
+ parameters = {
13
+ 'db' : 'iquod.db' ,
14
+ 'table' : 'unit'
15
+ }
13
16
14
17
def setUp (self ):
15
18
# refresh this table every test
@@ -25,18 +28,18 @@ def test_ICDC_level_order_simple(self):
25
28
[2.0 , - 1.0 , 1.0 ],
26
29
uid = 8888 )
27
30
qc = ICDC .test (p , self .parameters )
28
- nlevels , zr , tr = ICDC .reordered_data (p )
29
- zreverted = ICDC .revert_order (p , zr )
30
- zreverted_truth = np .ma .array ([2.0 , - 1.0 , 1.0 ],
31
+ nlevels , zr , tr = ICDC .reordered_data (p , self . parameters )
32
+ zreverted = ICDC .revert_order (p , zr , self . parameters )
33
+ zreverted_truth = np .ma .array ([2.0 , - 1.0 , 1.0 ],
31
34
mask = [False , True , False ])
32
-
35
+
33
36
assert np .array_equal (qc , [False , True , False ]), 'QC error'
34
37
assert nlevels == 2 , 'Subsetting of levels incorrect'
35
38
assert np .array_equal (zr , [1.0 , 2.0 ]), 'Depth reorder failed'
36
39
assert np .array_equal (tr , [3.0 , 1.0 ]), 'Temperature reorder failed'
37
- assert np .array_equal (zreverted .data [zreverted .mask == False ],
40
+ assert np .array_equal (zreverted .data [zreverted .mask == False ],
38
41
zreverted_truth .data [zreverted_truth .mask == False ]), 'Revert data failed'
39
- assert np .array_equal (zreverted .mask ,
42
+ assert np .array_equal (zreverted .mask ,
40
43
zreverted_truth .mask ), 'Revert data failed'
41
44
42
45
def test_ICDC_level_order (self ):
@@ -64,7 +67,7 @@ def test_ICDC_level_order(self):
64
67
assert np .array_equal (qc , qctruth ), 'Example {} QC wrong' .format (i + 1 )
65
68
66
69
# Check that the reordering is correct.
67
- nlevels , zr , tr = ICDC .reordered_data (p )
70
+ nlevels , zr , tr = ICDC .reordered_data (p , self . parameters )
68
71
assert np .array_equal (zr , ztruth ), 'Example {} zr wrong' .format (i + 1 )
69
72
assert np .array_equal (tr , ttruth ), 'Example {} tr wrong' .format (i + 1 )
70
73
0 commit comments