kopia lustrzana https://github.com/OpenDroneMap/ODM
65 wiersze
3.5 KiB
Python
65 wiersze
3.5 KiB
Python
|
import unittest
|
||
|
|
import os
|
||
|
|
from opendm.osfm import get_submodel_argv, get_submodel_args_dict
|
||
|
|
from opendm import config
|
||
|
|
|
||
|
|
class TestOSFM(unittest.TestCase):
|
||
|
|
def setUp(self):
|
||
|
|
pass
|
||
|
|
|
||
|
|
def test_get_submodel_argv(self):
|
||
|
|
# Base
|
||
|
|
args = config.config(["--project-path", "/datasets"])
|
||
|
|
|
||
|
|
self.assertEqual(get_submodel_argv(args)[1:],
|
||
|
|
['--orthophoto-cutline', '--dem-euclidean-map', '--skip-3dmodel'])
|
||
|
|
self.assertEqual(get_submodel_argv(args, "/submodels", "submodel_0000")[1:],
|
||
|
|
['--orthophoto-cutline', '--dem-euclidean-map', '--skip-3dmodel', '--project-path', '/submodels', 'submodel_0000'])
|
||
|
|
|
||
|
|
# Base + project name
|
||
|
|
args = config.config(["--project-path", "/datasets", "brighton"])
|
||
|
|
self.assertEqual(get_submodel_argv(args)[1:],
|
||
|
|
['--orthophoto-cutline', '--dem-euclidean-map', '--skip-3dmodel'])
|
||
|
|
self.assertEqual(get_submodel_argv(args, "/submodels", "submodel_0000")[1:],
|
||
|
|
['--orthophoto-cutline', '--dem-euclidean-map', '--skip-3dmodel', '--project-path', '/submodels', 'submodel_0000'])
|
||
|
|
|
||
|
|
# Project name + base
|
||
|
|
args = config.config(["brighton", "--project-path", "/datasets"])
|
||
|
|
self.assertEqual(get_submodel_argv(args)[1:],
|
||
|
|
['--orthophoto-cutline', '--dem-euclidean-map', '--skip-3dmodel'])
|
||
|
|
self.assertEqual(get_submodel_argv(args, "/submodels", "submodel_0000")[1:],
|
||
|
|
['--orthophoto-cutline', '--dem-euclidean-map', '--skip-3dmodel', '--project-path', '/submodels', 'submodel_0000'])
|
||
|
|
|
||
|
|
# Crop
|
||
|
|
args = config.config(["brighton", "--project-path", "/datasets", "--crop", "0"])
|
||
|
|
self.assertEqual(get_submodel_argv(args)[1:],
|
||
|
|
['--crop', '0.015625', '--orthophoto-cutline', '--dem-euclidean-map', '--skip-3dmodel'])
|
||
|
|
self.assertEqual(get_submodel_argv(args, "/submodels", "submodel_0000")[1:],
|
||
|
|
['--crop', '0.015625', '--orthophoto-cutline', '--dem-euclidean-map', '--skip-3dmodel', '--project-path', '/submodels', 'submodel_0000'])
|
||
|
|
|
||
|
|
# With sm-cluster, pc-csv and others
|
||
|
|
args = config.config(["--project-path", "/datasets", "--split", "200", "--pc-csv"])
|
||
|
|
self.assertEqual(get_submodel_argv(args)[1:],
|
||
|
|
['--orthophoto-cutline', '--dem-euclidean-map', '--skip-3dmodel'])
|
||
|
|
self.assertEqual(get_submodel_argv(args, "/submodels", "submodel_0000")[1:],
|
||
|
|
['--orthophoto-cutline', '--dem-euclidean-map', '--skip-3dmodel', '--project-path', '/submodels', 'submodel_0000'])
|
||
|
|
|
||
|
|
# Cameras JSON
|
||
|
|
args = config.config(["--project-path", "/datasets", "--cameras", os.path.join(os.path.dirname(os.path.realpath(__file__)), "assets", "sample.json")])
|
||
|
|
self.assertEqual(get_submodel_argv(args)[1:],
|
||
|
|
['--cameras', '{"test": "1"}', '--orthophoto-cutline', '--dem-euclidean-map', '--skip-3dmodel'])
|
||
|
|
|
||
|
|
# Camera JSON string
|
||
|
|
args = config.config(["--project-path", "/datasets", "--cameras", '{"test": "1"}'])
|
||
|
|
self.assertEqual(get_submodel_argv(args)[1:],
|
||
|
|
['--cameras', '{"test": "1"}', '--orthophoto-cutline', '--dem-euclidean-map', '--skip-3dmodel'])
|
||
|
|
|
||
|
|
def test_get_submodel_argv_dict(self):
|
||
|
|
# Base
|
||
|
|
args = config.config(["--project-path", "/datasets"])
|
||
|
|
|
||
|
|
self.assertEqual(get_submodel_args_dict(args),
|
||
|
|
{'orthophoto-cutline': True, 'skip-3dmodel': True, 'dem-euclidean-map': True})
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
unittest.main()
|