@@ -32,25 +32,25 @@ def testEdit(self):
32
32
ml = QgsVectorLayer ("Point?crs=epsg:4236&field=id:integer&field=value:double" ,
33
33
"test_data" , "memory" )
34
34
# Data as list of x, y, id, value
35
- assert ml .isValid ()
35
+ self . assertTrue ( ml .isValid () )
36
36
fields = ml .fields ()
37
37
38
38
# Check insert
39
39
with edit (ml ):
40
40
feat = QgsFeature (fields )
41
41
feat ['id' ] = 1
42
42
feat ['value' ] = 0.9
43
- assert ml .addFeature (feat )
43
+ self . assertTrue ( ml .addFeature (feat ) )
44
44
45
- assert ml .dataProvider ().getFeatures (). next ( )['value' ] == 0.9
45
+ self . assertEqual ( next ( ml .dataProvider ().getFeatures ())['value' ], 0.9 )
46
46
47
47
# Check update
48
48
with edit (ml ):
49
49
f = next (ml .getFeatures ())
50
50
f ['value' ] = 9.9
51
- assert ml .updateFeature (f )
51
+ self . assertTrue ( ml .updateFeature (f ) )
52
52
53
- assert ml .dataProvider ().getFeatures (). next ( )['value' ] == 9.9
53
+ self . assertEqual ( next ( ml .dataProvider ().getFeatures ())['value' ], 9.9 )
54
54
55
55
# Check for rollBack after exceptions
56
56
with self .assertRaises (NameError ):
@@ -59,16 +59,16 @@ def testEdit(self):
59
59
f ['value' ] = 3.8
60
60
crashycrash () # NOQA
61
61
62
- assert ml .dataProvider ().getFeatures (). next ( )['value' ] == 9.9
63
- assert ml .getFeatures (). next ( )['value' ] == 9.9
62
+ self . assertEqual ( next ( ml .dataProvider ().getFeatures ())['value' ], 9.9 )
63
+ self . assertEqual ( next ( ml .getFeatures ())['value' ], 9.9 )
64
64
65
65
# Check for `as`
66
66
with edit (ml ) as l :
67
67
f = next (l .getFeatures ())
68
68
f ['value' ] = 10
69
- assert l .updateFeature (f )
69
+ self . assertTrue ( l .updateFeature (f ) )
70
70
71
- assert ml .dataProvider ().getFeatures (). next ( )['value' ] == 10
71
+ self . assertEqual ( next ( ml .dataProvider ().getFeatures ())['value' ], 10 )
72
72
73
73
# Check that we get a QgsEditError exception when the commit fails
74
74
with self .assertRaises (QgsEditError ):
0 commit comments