@@ -652,7 +652,7 @@ def test_at_time_axis(self):
652
652
assert_frame_equal (result , expected )
653
653
654
654
ts .columns = rng
655
- expected = ts .loc [:,indices ]
655
+ expected = ts .loc [:, indices ]
656
656
result = ts .at_time ('9:30' , axis = 1 )
657
657
assert_frame_equal (result , expected )
658
658
@@ -724,11 +724,10 @@ def test_between_time_raises(self):
724
724
725
725
def test_between_time_axis (self ):
726
726
rng = date_range ('1/1/2000' , periods = 100 , freq = '10min' )
727
- blank = np .arange (0 ,len (rng ))
727
+ blank = np .arange (0 , len (rng ))
728
728
stime , etime = ('08:00:00' , '09:00:00' )
729
729
dimn = (len (rng ), len (rng ))
730
730
exp_len = 7
731
-
732
731
for time_index , time_col in product ([True , False ], [True , False ]):
733
732
if time_index :
734
733
index = rng
@@ -746,15 +745,13 @@ def test_between_time_axis(self):
746
745
assert len (ts .between_time (stime , etime , axis = 0 )) == exp_len
747
746
else :
748
747
pytest .raises (TypeError , ts .between_time , stime , etime )
749
- pytest .raises (TypeError , ts .between_time , stime , etime ,
750
- axis = 0 )
748
+ pytest .raises (TypeError , ts .between_time , stime , etime , axis = 0 )
751
749
752
750
if time_col :
753
751
selected = ts .between_time (stime , etime , axis = 1 ).columns
754
752
assert len (selected ) == exp_len
755
753
else :
756
- pytest .raises (TypeError , ts .between_time , stime , etime ,
757
- axis = 1 )
754
+ pytest .raises (TypeError , ts .between_time , stime , etime , axis = 1 )
758
755
759
756
def test_operation_on_NaT (self ):
760
757
# Both NaT and Timestamp are in DataFrame.
0 commit comments