Meshroom/tests/test_nodeCommandLineFormatting.py
2024-09-12 10:50:13 +02:00

78 lines
3.2 KiB
Python

#!/usr/bin/env python
# coding:utf-8
import meshroom.multiview
from meshroom.core.graph import Graph
def test_formatting_listOfFiles():
meshroom.core.initNodes()
inputImages = ['/non/existing/fileA', '/non/existing/with space/fileB']
graph = Graph('')
n1 = graph.addNewNode('CameraInit')
n1.viewpoints.extend([{'path': image} for image in inputImages])
# viewId, poseId, path, intrinsicId, rigId, subPoseId, metadata
assert n1.viewpoints.getValueStr() == \
'-1 -1 "/non/existing/fileA" -1 -1 -1 "" -1 -1 "/non/existing/with space/fileB" -1 -1 -1 ""'
graph = Graph('')
n1 = graph.addNewNode('ImageMatching')
assert n1.featuresFolders.getValueStr() == ''
n1.featuresFolders.extend("single value with space")
assert n1.featuresFolders.getValueStr() == '"single value with space"'
n1.featuresFolders.resetToDefaultValue()
assert n1.featuresFolders.getValueStr() == ''
n1.featuresFolders.extend(inputImages)
assert n1.featuresFolders.getValueStr() == '"/non/existing/fileA" "/non/existing/with space/fileB"'
n1._buildCmdVars() # prepare vars for command line creation
# and check some values
name = 'featuresFolders'
assert n1._cmdVars[name + 'Value'] == '/non/existing/fileA /non/existing/with space/fileB'
def test_formatting_strings():
meshroom.core.initNodes()
graph = Graph('')
n1 = graph.addNewNode('ImageMatching')
name = 'weights'
assert n1.weights.getValueStr() == '""' # Empty string should generate empty quotes
assert n1._cmdVars[name + 'Value'] == ''
name = 'method'
assert n1.method.getValueStr() == '"SequentialAndVocabularyTree"'
assert n1._cmdVars[name + 'Value'] == 'SequentialAndVocabularyTree'
n2 = graph.addNewNode('ImageMatching')
n2._buildCmdVars() # prepare vars for command line creation
name = 'featuresFolders'
assert n2._cmdVars[name + 'Value'] == '', 'Empty list should become fully empty'
n2.featuresFolders.extend('')
n2._buildCmdVars() # prepare vars for command line creation
assert n2.featuresFolders.getValueStr() == '""', 'A list with one empty string should generate empty quotes'
assert n2._cmdVars[name + 'Value'] == '', 'The Value is always only the value, so empty here'
n2.featuresFolders.extend('')
n2._buildCmdVars() # prepare vars for command line creation
assert n2.featuresFolders.getValueStr() == '"" ""', 'A list with 2 empty strings should generate quotes'
assert n2._cmdVars[name + 'Value'] == ' ', \
'The Value is always only the value, so 2 empty with the space separator in the middle'
def test_formatting_groups():
meshroom.core.initNodes()
graph = Graph('')
n3 = graph.addNewNode('ImageProcessing')
n3._buildCmdVars() # prepare vars for command line creation
name = 'sharpenFilter'
assert n3.sharpenFilter.getValueStr() == '"False:3:1.0:0.0"'
assert n3._cmdVars[name + 'Value'] == 'False:3:1.0:0.0', 'The Value is always only the value, so no quotes'
name = 'fillHoles'
assert n3._cmdVars[name + 'Value'] == 'False', 'Booleans'
name = 'noiseFilter'
assert n3.noiseFilter.getValueStr() == '"False:uniform:0.0:1.0:True"'
assert n3._cmdVars[name + 'Value'] == 'False:uniform:0.0:1.0:True'