Add import pytest in some files in integration_tests (#286)

Add import pytest in some files in integration_tests (#286) #375