@@ -22,13 +22,13 @@ def mix_abc() -> Dict[str, List[Union[float, str]]]:
22
22
23
23
24
24
class TestDataFrameReplace :
25
- def test_replace_inplace (self , datetime_frame , float_string_frame ):
26
- datetime_frame ["A" ][:5 ] = np .nan
27
- datetime_frame ["A" ][- 5 :] = np .nan
25
+ def test_replace_inplace (self , float_frame_dt_index , float_string_frame ):
26
+ float_frame_dt_index ["A" ][:5 ] = np .nan
27
+ float_frame_dt_index ["A" ][- 5 :] = np .nan
28
28
29
- tsframe = datetime_frame .copy ()
29
+ tsframe = float_frame_dt_index .copy ()
30
30
tsframe .replace (np .nan , 0 , inplace = True )
31
- tm .assert_frame_equal (tsframe , datetime_frame .fillna (0 ))
31
+ tm .assert_frame_equal (tsframe , float_frame_dt_index .fillna (0 ))
32
32
33
33
# mixed type
34
34
mf = float_string_frame
@@ -39,9 +39,9 @@ def test_replace_inplace(self, datetime_frame, float_string_frame):
39
39
expected = float_string_frame .fillna (value = 0 )
40
40
tm .assert_frame_equal (result , expected )
41
41
42
- tsframe = datetime_frame .copy ()
42
+ tsframe = float_frame_dt_index .copy ()
43
43
tsframe .replace ([np .nan ], [0 ], inplace = True )
44
- tm .assert_frame_equal (tsframe , datetime_frame .fillna (0 ))
44
+ tm .assert_frame_equal (tsframe , float_frame_dt_index .fillna (0 ))
45
45
46
46
def test_regex_replace_scalar (self , mix_ab ):
47
47
obj = {"a" : list ("ab.." ), "b" : list ("efgh" )}
@@ -582,17 +582,17 @@ def test_replace_regex_metachar(self, metachar):
582
582
expected = DataFrame ({"a" : ["paren" , "else" ]})
583
583
tm .assert_frame_equal (result , expected )
584
584
585
- def test_replace (self , datetime_frame ):
586
- datetime_frame ["A" ][:5 ] = np .nan
587
- datetime_frame ["A" ][- 5 :] = np .nan
585
+ def test_replace (self , float_frame_dt_index ):
586
+ float_frame_dt_index ["A" ][:5 ] = np .nan
587
+ float_frame_dt_index ["A" ][- 5 :] = np .nan
588
588
589
- zero_filled = datetime_frame .replace (np .nan , - 1e8 )
590
- tm .assert_frame_equal (zero_filled , datetime_frame .fillna (- 1e8 ))
591
- tm .assert_frame_equal (zero_filled .replace (- 1e8 , np .nan ), datetime_frame )
589
+ zero_filled = float_frame_dt_index .replace (np .nan , - 1e8 )
590
+ tm .assert_frame_equal (zero_filled , float_frame_dt_index .fillna (- 1e8 ))
591
+ tm .assert_frame_equal (zero_filled .replace (- 1e8 , np .nan ), float_frame_dt_index )
592
592
593
- datetime_frame ["A" ][:5 ] = np .nan
594
- datetime_frame ["A" ][- 5 :] = np .nan
595
- datetime_frame ["B" ][:5 ] = - 1e8
593
+ float_frame_dt_index ["A" ][:5 ] = np .nan
594
+ float_frame_dt_index ["A" ][- 5 :] = np .nan
595
+ float_frame_dt_index ["B" ][:5 ] = - 1e8
596
596
597
597
# empty
598
598
df = DataFrame (index = ["a" , "b" ])
@@ -792,30 +792,30 @@ def test_replace_simple_nested_dict_with_nonexistent_value(self):
792
792
result = df .replace ({"col" : {- 1 : "-" , 1 : "a" , 4 : "b" }})
793
793
tm .assert_frame_equal (expected , result )
794
794
795
- def test_replace_value_is_none (self , datetime_frame ):
796
- orig_value = datetime_frame .iloc [0 , 0 ]
797
- orig2 = datetime_frame .iloc [1 , 0 ]
795
+ def test_replace_value_is_none (self , float_frame_dt_index ):
796
+ orig_value = float_frame_dt_index .iloc [0 , 0 ]
797
+ orig2 = float_frame_dt_index .iloc [1 , 0 ]
798
798
799
- datetime_frame .iloc [0 , 0 ] = np .nan
800
- datetime_frame .iloc [1 , 0 ] = 1
799
+ float_frame_dt_index .iloc [0 , 0 ] = np .nan
800
+ float_frame_dt_index .iloc [1 , 0 ] = 1
801
801
802
- result = datetime_frame .replace (to_replace = {np .nan : 0 })
803
- expected = datetime_frame .T .replace (to_replace = {np .nan : 0 }).T
802
+ result = float_frame_dt_index .replace (to_replace = {np .nan : 0 })
803
+ expected = float_frame_dt_index .T .replace (to_replace = {np .nan : 0 }).T
804
804
tm .assert_frame_equal (result , expected )
805
805
806
- result = datetime_frame .replace (to_replace = {np .nan : 0 , 1 : - 1e8 })
807
- tsframe = datetime_frame .copy ()
806
+ result = float_frame_dt_index .replace (to_replace = {np .nan : 0 , 1 : - 1e8 })
807
+ tsframe = float_frame_dt_index .copy ()
808
808
tsframe .iloc [0 , 0 ] = 0
809
809
tsframe .iloc [1 , 0 ] = - 1e8
810
810
expected = tsframe
811
811
tm .assert_frame_equal (expected , result )
812
- datetime_frame .iloc [0 , 0 ] = orig_value
813
- datetime_frame .iloc [1 , 0 ] = orig2
812
+ float_frame_dt_index .iloc [0 , 0 ] = orig_value
813
+ float_frame_dt_index .iloc [1 , 0 ] = orig2
814
814
815
- def test_replace_for_new_dtypes (self , datetime_frame ):
815
+ def test_replace_for_new_dtypes (self , float_frame_dt_index ):
816
816
817
817
# dtypes
818
- tsframe = datetime_frame .copy ().astype (np .float32 )
818
+ tsframe = float_frame_dt_index .copy ().astype (np .float32 )
819
819
tsframe ["A" ][:5 ] = np .nan
820
820
tsframe ["A" ][- 5 :] = np .nan
821
821
0 commit comments