<s> import unittest <EOL> from vprof import runtime_profile <EOL> try : <EOL> import mock <EOL> except ImportError : <EOL> from unittest import mock <EOL> _RUN_STATS = { <EOL> ( '<STR_LIT>' , <NUM_LIT:1> , '<STR_LIT>' ) : ( <NUM_LIT:1> , <NUM_LIT:10> , <NUM_LIT> , <NUM_LIT> , { <EOL> ( '<STR_LIT>' , <NUM_LIT:1> , '<STR_LIT>' ) : ( <NUM_LIT:1> , <NUM_LIT:1> , <NUM_LIT> , <NUM_LIT> ) , <EOL> ( '<STR_LIT>' , <NUM_LIT:1> , '<STR_LIT>' ) : ( <NUM_LIT:9> , <NUM_LIT:1> , <NUM_LIT> , <NUM_LIT> ) <EOL> } ) , <EOL> ( '<STR_LIT>' , <NUM_LIT:1> , '<STR_LIT>' ) : ( <NUM_LIT:1> , <NUM_LIT:1> , <NUM_LIT> , <NUM_LIT> , { } ) , <EOL> ( '<STR_LIT>' , <NUM_LIT:0> , '<STR_LIT>' ) : ( <NUM_LIT:1> , <NUM_LIT:1> , <NUM_LIT> , <NUM_LIT> , { <EOL> ( '<STR_LIT>' , <NUM_LIT:1> , '<STR_LIT>' ) : ( <NUM_LIT:1> , <NUM_LIT:1> , <NUM_LIT> , <NUM_LIT> ) <EOL> } ) <EOL> } <EOL> _CALLEES = { <EOL> ( '<STR_LIT>' , <NUM_LIT:1> , '<STR_LIT>' ) : [ <EOL> ( '<STR_LIT>' , <NUM_LIT:1> , '<STR_LIT>' ) , <EOL> ( '<STR_LIT>' , <NUM_LIT:0> , '<STR_LIT>' ) , <EOL> ] , <EOL> ( '<STR_LIT>' , <NUM_LIT:1> , '<STR_LIT>' ) : [ <EOL> ( '<STR_LIT>' , <NUM_LIT:1> , '<STR_LIT>' ) <EOL> ] <EOL> } <EOL> _CALL_GRAPH = { <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : <NUM_LIT:1> , <EOL> '<STR_LIT>' : <NUM_LIT:1> , <EOL> '<STR_LIT>' : <NUM_LIT> , <EOL> '<STR_LIT>' : <NUM_LIT> , <EOL> '<STR_LIT>' : <NUM_LIT:1> , <EOL> '<STR_LIT>' : [ <EOL> { '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : <NUM_LIT:1> , <EOL> '<STR_LIT>' : <NUM_LIT:10> , <EOL> '<STR_LIT>' : <NUM_LIT> , <EOL> '<STR_LIT>' : <NUM_LIT> , <EOL> '<STR_LIT>' : <NUM_LIT:1> , <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : [ ] } , <EOL> { '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : <NUM_LIT:1> , <EOL> '<STR_LIT>' : <NUM_LIT:1> , <EOL> '<STR_LIT>' : <NUM_LIT> , <EOL> '<STR_LIT>' : <NUM_LIT> , <EOL> '<STR_LIT>' : <NUM_LIT:0> , <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : [ ] } <EOL> ] <EOL> } <EOL> class RuntimeProfileUnittest ( unittest . TestCase ) : <EOL> def setUp ( self ) : <EOL> self . _profile = object . __new__ ( runtime_profile . RuntimeProfile ) <EOL> def testBuildCallees ( self ) : <EOL> self . assertDictEqual ( <EOL> dict ( self . _profile . _build_callees ( _RUN_STATS ) ) , _CALLEES ) <EOL> def testTransformStats ( self ) : <EOL> stats = mock . MagicMock ( ) <EOL> stats . stats = _RUN_STATS <EOL> self . assertDictEqual ( <EOL> self . _profile . _transform_stats ( stats ) , _CALL_GRAPH ) </s>
