Coverage for /pythoncovmergedfiles/medio/medio/src/fuzz_to_datetime.py: 52%

1###### Coverage stub 

2import atexit 

3import coverage 

4cov = coverage.coverage(data_file='.coverage', cover_pylib=True) 

5cov.start() 

6# Register an exist handler that will print coverage 

7def exit_handler(): 

8 cov.stop() 

9 cov.save() 

10atexit.register(exit_handler) 

11####### End of coverage stub 

12#!/usr/bin/python3 

13# Copyright 2023 Google LLC 

14# 

15# Licensed under the Apache License, Version 2.0 (the "License"); 

16# you may not use this file except in compliance with the License. 

17# You may obtain a copy of the License at 

18# 

19# http://www.apache.org/licenses/LICENSE-2.0 

20# 

21# Unless required by applicable law or agreed to in writing, software 

22# distributed under the License is distributed on an "AS IS" BASIS, 

23# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 

24# See the License for the specific language governing permissions and 

25# limitations under the License. 

26"""Targets the pandas to_datetime function.""" 

27 

28import sys 

29import atheris 

30import pandas as pd 

31 

32from pandas.errors import ( 

33 ParserError, 

34 OutOfBoundsDatetime 

35) 

36 

37 

38def TestOneInput(data): 

39 fdp = atheris.FuzzedDataProvider(data) 

40 

41 try: 

42 input_str = fdp.ConsumeUnicodeNoSurrogates(100) 

43 errors = fdp.PickValueInList(['raise', 'coerce', 'ignore', fdp.ConsumeUnicodeNoSurrogates(20)]) 

44 unit = fdp.PickValueInList([None, 'ns', 'us', 'ms', 's', 'D', 'M', 'Y', fdp.ConsumeUnicodeNoSurrogates(20)]) 

45 origin = fdp.PickValueInList(['unix', 'julian', 'epoch', fdp.ConsumeUnicodeNoSurrogates(20)]) 

46 dayfirst = fdp.ConsumeBool() 

47 yearfirst = fdp.ConsumeBool() 

48 exact = fdp.ConsumeBool() 

49 cache = fdp.ConsumeBool() 

50 

51 pd.to_datetime( 

52 arg=input_str, 

53 errors=errors, 

54 unit=unit, 

55 origin=origin, 

56 dayfirst=dayfirst, 

57 yearfirst=yearfirst, 

58 exact=exact, 

59 cache=cache 

60 ) 

61 

62 except ( 

63 ParserError, # When parsing a date from string fails. 

64 ValueError, # When another datetime conversion error happens. 

65 OutOfBoundsDatetime # catching attempts to create a DatetimeIndex, which may raise from cast() 

66 ): 

67 pass 

68 

69 

70def main(): 

71 atheris.Setup(sys.argv, TestOneInput) 

72 atheris.instrument_all() 

73 atheris.Fuzz() 

74 

75 

76if __name__ == "__main__": 

77 main()