mirror of
https://github.com/opencv/opencv.git
synced 2024-12-12 15:19:11 +08:00
51 lines
1.1 KiB
Python
Executable File
51 lines
1.1 KiB
Python
Executable File
#! /usr/bin/env python
|
|
"""
|
|
This script will test highgui's image saving functionality
|
|
"""
|
|
|
|
# name if this test and it's requirements
|
|
TESTNAME = "cvSaveImage"
|
|
REQUIRED = ["cvLoadImagejpg"]
|
|
|
|
#needed for sys.exit(int), filehandling and .works file checks
|
|
import os
|
|
import sys
|
|
import works
|
|
|
|
# path to imagefiles we need
|
|
PREFIX=os.path.join(os.environ["srcdir"],"../../opencv_extra/testdata/python/images/")
|
|
|
|
# delete old .works file and check requirements
|
|
if not works.check_files(REQUIRED,TESTNAME):
|
|
sys.exit(77)
|
|
|
|
# import the necessary things for OpenCV
|
|
from highgui import *
|
|
from cv import *
|
|
|
|
# our temporary test file
|
|
file_name = "./highgui_testfile.bmp"
|
|
|
|
# try to load an image from a file
|
|
image = cvLoadImage(PREFIX+"baboon.jpg")
|
|
|
|
# if the returned object is not Null, loading was successful.
|
|
if image==0:
|
|
print "(INFO) Couldn't load test image. Skipping rest of this test."
|
|
sys.exit(77)
|
|
|
|
res = cvSaveImage("./highgui_testfile.bmp", image)
|
|
|
|
if res == 0:
|
|
print "(ERROR) Couldn't save image to '"+file_name+"'."
|
|
sys.exit(1)
|
|
|
|
# remove temporary file
|
|
os.remove(file_name)
|
|
|
|
# create flag file
|
|
works.set_file(TESTNAME)
|
|
|
|
# return 0 ('PASS')
|
|
sys.exit(0)
|