Abstract: A computer-implemented method, a computer program product, and a computer system for using computer vision to automatically generate a unit test case. A computer converts a text file of a source code to an image file of the source code. A computer inputs the image file to a neural network having been trained with source code images. The neural network identifies elements in the image file. The neural network generates a resulting image including identified elements in the image file. A computer extracts the identified elements from the resulting image. A computer generates a text file including a key value map of the identified elements. A computer generates a scaffold of a unit test case, based on the key value map.