@@ -47,25 +47,26 @@ def test_update_value(self):
4747
4848 def test_add_points (self ):
4949 #with pytest.raises(NotImplementedError):
50- assert len (self .memory_db .df )== 10
50+ assert len (self .memory_db .getDataValuesDF (). index )== 10
5151 point = [('-9999' , None , datetime .datetime (2011 , 3 , 25 , 0 , 0 ), '-7' , datetime .datetime (2015 , 3 , 25 , 7 , 0 ), None ,
5252 None , u'nc' , None , None , self .series .site_id , self .series .variable_id , self .series .method_id ,
5353 self .series .source_id , self .series .quality_control_level_id )]
5454 self .memory_db .addPoints (point )
5555 dvs = self .memory_db .getDataValuesDF ()
5656
57- assert len (dvs ) == 11
58- assert dvs ["DataValue" ][- 1 ] == - 9999
57+ assert len (dvs . index ) == 11
58+ assert dvs ["DataValue" ][0 ] == - 9999
5959
6060 def test_update_flag (self ):
61- self .memory_db .updateFlag (self .sdate , '50' )
61+ self .memory_db .updateFlag ([ self .sdate ] , '50' )
6262 dvs = self .memory_db .getDataValuesDF ()
6363 assert dvs ["QualifierID" ][0 ] == '50'
6464
6565
6666 def test_delete_points (self ):
67- stlen = len (self .memory_db .getDataValuesDF ())
68- self .memory_db .delete ([1 , 2 , 3 , 4 , 5 , 6 , 7 , 8 , 9 , 10 ])
67+ stlen = len (self .memory_db .df .index )
68+
69+ self .memory_db .delete (self .memory_db .df ["LocalDateTime" ].tolist ()[0 :10 ])
6970 dvs = self .memory_db .getDataValuesDF ()
70- assert len (dvs ) == stlen - 10
71+ assert len (dvs . index ) == stlen - 10
7172
0 commit comments