vendor: OpenCV 5.0.0 snapshot at 40738fb16ceddb5fb3fea747585f7ce6abb0605b
This commit is contained in:
@@ -0,0 +1,87 @@
|
||||
#!/usr/bin/env python
|
||||
""""Core serialization tests."""
|
||||
import tempfile
|
||||
import os
|
||||
import cv2 as cv
|
||||
import numpy as np
|
||||
from tests_common import NewOpenCVTests, unittest
|
||||
|
||||
|
||||
class persistence_test(NewOpenCVTests):
|
||||
def test_yml_rw(self):
|
||||
fd, fname = tempfile.mkstemp(prefix="opencv_python_persistence_", suffix=".yml")
|
||||
os.close(fd)
|
||||
|
||||
# Writing ...
|
||||
expected = np.array([[[0, 1, 2, 3, 4]]])
|
||||
expected_str = ("Hello", "World", "!")
|
||||
fs = cv.FileStorage(fname, cv.FILE_STORAGE_WRITE)
|
||||
fs.write("test", expected)
|
||||
fs.write("strings", expected_str)
|
||||
fs.release()
|
||||
|
||||
# Reading ...
|
||||
fs = cv.FileStorage(fname, cv.FILE_STORAGE_READ)
|
||||
root = fs.getFirstTopLevelNode()
|
||||
self.assertEqual(root.name(), "test")
|
||||
|
||||
test = fs.getNode("test")
|
||||
self.assertEqual(test.empty(), False)
|
||||
self.assertEqual(test.name(), "test")
|
||||
self.assertEqual(test.type(), cv.FILE_NODE_MAP)
|
||||
self.assertEqual(test.isMap(), True)
|
||||
actual = test.mat()
|
||||
self.assertEqual(actual.shape, expected.shape)
|
||||
self.assertEqual(np.array_equal(expected, actual), True)
|
||||
|
||||
strings = fs.getNode("strings")
|
||||
self.assertEqual(strings.isSeq(), True)
|
||||
self.assertEqual(strings.size(), len(expected_str))
|
||||
self.assertEqual(all(strings.at(i).isString() for i in range(strings.size())), True)
|
||||
self.assertSequenceEqual([strings.at(i).string() for i in range(strings.size())], expected_str)
|
||||
fs.release()
|
||||
|
||||
os.remove(fname)
|
||||
|
||||
def test_yml_python_interop(self):
|
||||
try:
|
||||
import yaml
|
||||
except:
|
||||
raise unittest.SkipTest('Pyyml is not available for interop test')
|
||||
|
||||
ref_data = {
|
||||
'int_value': 42,
|
||||
"bool_value": True,
|
||||
"float_value": 3.1415926,
|
||||
"int64_value": 2147483647 + 1024, # C++ INT_MAX + 1024
|
||||
"string_value": "opencv"
|
||||
}
|
||||
|
||||
fd, test_file_name = tempfile.mkstemp(prefix="opencv_python_persistence_", suffix=".yml")
|
||||
os.close(fd)
|
||||
|
||||
with open(test_file_name, 'w') as ff:
|
||||
yaml.dump(ref_data, ff)
|
||||
|
||||
# Notice: no cv.FileStorage_FORMAT_YAML flag needed now thanks to the C++ fix!
|
||||
fs = cv.FileStorage(test_file_name, cv.FILE_STORAGE_READ)
|
||||
self.assertTrue(fs.isOpened())
|
||||
|
||||
node = fs.getNode('int_value')
|
||||
self.assertTrue(node.isInt())
|
||||
self.assertEqual(42, int(node.real()))
|
||||
|
||||
node = fs.getNode('int64_value')
|
||||
self.assertTrue(node.isInt())
|
||||
self.assertEqual(2147483647 + 1024, int(node.real()))
|
||||
|
||||
node = fs.getNode('float_value')
|
||||
self.assertTrue(node.isReal())
|
||||
self.assertEqual(3.1415926, node.real())
|
||||
|
||||
node = fs.getNode('string_value')
|
||||
self.assertTrue(node.isString())
|
||||
self.assertEqual("opencv", node.string())
|
||||
|
||||
fs.release()
|
||||
os.remove(test_file_name)
|
||||
Reference in New Issue
Block a user