Coverage for /pythoncovmergedfiles/medio/medio/src/onnx/onnx/fuzz/fuzz_model_loader.py: 43%

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# Copyright (c) ONNX Project Contributors 

13# SPDX-License-Identifier: Apache-2.0 

14 

15import sys 

16 

17import atheris 

18 

19with atheris.instrument_imports(): 

20 import onnx 

21 

22 

23def TestOneInput(data): 

24 try: 

25 model = onnx.load_model_from_string(data) 

26 _ = len(model.graph.node) 

27 _ = len(model.graph.input) 

28 _ = len(model.graph.output) 

29 onnx.checker.check_model(model) 

30 except Exception: 

31 return 

32 

33 

34def main(): 

35 atheris.instrument_all() 

36 atheris.Setup(sys.argv, TestOneInput, enable_python_coverage=True) 

37 atheris.Fuzz() 

38 

39 

40if __name__ == "__main__": 

41 main()