mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-06-06 00:43:52 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

test: FileStorage format regression test

Browse Source
This commit is contained in:
Alexander Alekhin 2017-07-20 19:56:47 +03:00
parent 9313978f61
commit 5bc291937f
3 changed files with 67 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
modules/core/include/opencv2/core/persistence.hpp
View File

@ -462,6 +462,11 @@ public:
*/
static String getDefaultObjectName(const String& filename);
/** @brief Returns the current format.
* @returns The current format, see FileStorage::Mode
*/
CV_WRAP int getFormat() const;
Ptr<CvFileStorage> fs; //!< the underlying C FileStorage structure
String elname; //!< the currently written element
std::vector<char> structs; //!< the stack of written structures

6
modules/core/src/persistence.cpp
View File

@ -6920,6 +6920,12 @@ FileNode FileStorage::root(int streamidx) const
return isOpened() ? FileNode(fs, cvGetRootFileNode(fs, streamidx)) : FileNode();
}
int FileStorage::getFormat() const
{
CV_Assert(!fs.empty());
return fs->fmt & FORMAT_MASK;
}
FileStorage& operator << (FileStorage& fs, const String& str)
{
CV_TRACE_REGION_VERBOSE();

56
modules/core/test/test_io.cpp
View File

@ -1236,3 +1236,59 @@ TEST(Core_InputOutput, FileStorage_DMatch_vector_vector)
}
}
}
TEST(Core_InputOutput, FileStorage_format_xml)
{
FileStorage fs;
fs.open("opencv_storage.xml", FileStorage::WRITE | FileStorage::MEMORY);
EXPECT_EQ(FileStorage::FORMAT_XML, fs.getFormat());
}
TEST(Core_InputOutput, FileStorage_format_xml_gz)
{
FileStorage fs;
fs.open("opencv_storage.xml.gz", FileStorage::WRITE | FileStorage::MEMORY);
EXPECT_EQ(FileStorage::FORMAT_XML, fs.getFormat());
}
TEST(Core_InputOutput, FileStorage_format_json)
{
FileStorage fs;
fs.open("opencv_storage.json", FileStorage::WRITE | FileStorage::MEMORY);
EXPECT_EQ(FileStorage::FORMAT_JSON, fs.getFormat());
}
TEST(Core_InputOutput, FileStorage_format_json_gz)
{
FileStorage fs;
fs.open("opencv_storage.json.gz", FileStorage::WRITE | FileStorage::MEMORY);
EXPECT_EQ(FileStorage::FORMAT_JSON, fs.getFormat());
}
TEST(Core_InputOutput, FileStorage_format_yaml)
{
FileStorage fs;
fs.open("opencv_storage.yaml", FileStorage::WRITE | FileStorage::MEMORY);
EXPECT_EQ(FileStorage::FORMAT_YAML, fs.getFormat());
}
TEST(Core_InputOutput, FileStorage_format_yaml_gz)
{
FileStorage fs;
fs.open("opencv_storage.yaml.gz", FileStorage::WRITE | FileStorage::MEMORY);
EXPECT_EQ(FileStorage::FORMAT_YAML, fs.getFormat());
}
TEST(Core_InputOutput, FileStorage_format_yml)
{
FileStorage fs;
fs.open("opencv_storage.yml", FileStorage::WRITE | FileStorage::MEMORY);
EXPECT_EQ(FileStorage::FORMAT_YAML, fs.getFormat());
}
TEST(Core_InputOutput, FileStorage_format_yml_gz)
{
FileStorage fs;
fs.open("opencv_storage.yml.gz", FileStorage::WRITE | FileStorage::MEMORY);
EXPECT_EQ(FileStorage::FORMAT_YAML, fs.getFormat());
}