File tree 2 files changed +12
-15
lines changed
2 files changed +12
-15
lines changed Original file line number Diff line number Diff line change @@ -583,7 +583,6 @@ def test_get_indexer(self):
583
583
def test_reasonable_keyerror (self ):
584
584
# GH#1062
585
585
index = DatetimeIndex (['1/3/2000' ])
586
- try :
586
+ with pytest . raises ( KeyError ) as excinfo :
587
587
index .get_loc ('1/1/2000' )
588
- except KeyError as e :
589
- assert '2000' in str (e )
588
+ assert '2000' in str (excinfo .value )
Original file line number Diff line number Diff line change 23
23
24
24
class CParserTests (object ):
25
25
26
- def test_buffer_overflow (self ):
26
+ @pytest .mark .parametrize (
27
+ 'malf' ,
28
+ ['1\r 1\r 1\r 1\r 1\r ' ,
29
+ '1\r 1\r 1\r 1\r 1\r 11\r ' ,
30
+ '1\r 1\r 1\r 1\r 1\r 11\r 1\r ' ],
31
+ ids = ['words pointer' , 'stream pointer' , 'lines pointer' ])
32
+ def test_buffer_overflow (self , malf ):
27
33
# see gh-9205: test certain malformed input files that cause
28
34
# buffer overflows in tokenizer.c
29
-
30
- malfw = "1\r 1\r 1\r 1\r 1\r " # buffer overflow in words pointer
31
- malfs = "1\r 1\r 1\r 1\r 1\r 11\r " # buffer overflow in stream pointer
32
- malfl = "1\r 1\r 1\r 1\r 1\r 11\r 1\r " # buffer overflow in lines pointer
33
-
34
35
cperr = 'Buffer overflow caught - possible malformed input file.'
35
-
36
- for malf in (malfw , malfs , malfl ):
37
- try :
38
- self .read_table (StringIO (malf ))
39
- except Exception as err :
40
- assert cperr in str (err )
36
+ with pytest .raises (pd .errors .ParserError ) as excinfo :
37
+ self .read_table (StringIO (malf ))
38
+ assert cperr in str (excinfo .value )
41
39
42
40
def test_buffer_rd_bytes (self ):
43
41
# see gh-12098: src->buffer in the C parser can be freed twice leading
You can’t perform that action at this time.
0 commit comments