How to capture the screenshot of the mobile app during unit test case execution in iOS?

Question

I\'m writing unit test cases for my iOS swift app. I just wanted to know is it possible to create screenshot of the device screen and store it as image in t