from unittest import main as unittest_main, TestCase
from ittapi_native_mock import patch as ittapi_native_patch
import ittapi.compat
class PT_RegionTests(TestCase):
@ittapi_native_patch('PT_Region')
def test_pt_region_create_call(self, pt_region_mock):
s='my region'
ittapi.compat.pt_region_create(s)
pt_region_mock.assert_called_once_with(s)
@ittapi_native_patch('PT_Region')
@ittapi_native_patch('pt_region_begin')
@ittapi_native_patch('pt_region_end')
def test_pt_region_begin_call(self, pt_region_mock, pt_region_begin_mock, pt_region_end_mock):
region_name='my region'
pt_region=ittapi.compat.pt_region_create(region_name)
pt_region_mock.assert_called_once_with(region_name)
ittapi.compat.pt_region_begin(pt_region)
pt_region_begin_mock.assert_called_once_with(pt_region)
ittapi.compat.pt_region_end(pt_region)
pt_region_end_mock.assert_called_once_with(pt_region)
if __name__ == '__main__':
unittest_main()