1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
|
"""
Collection of unit tests for the SfMDataIO class.
"""
import os
from pyalicevision import sfmDataIO, sfmData
from ..constants import SFMDATA_PATH, IMAGE_PATH, VIEW_ID, INTRINSIC_ID, POSE_ID, \
IMAGE_WIDTH, IMAGE_HEIGHT, RIG_ID, SUBPOSE_ID, METADATA
##################
### List of functions:
# - bool validIds(SfMData& sfmData, ESfMData partFlag) => DONE
# - bool load(SfMData& sfmData, string& filename, ESfMData partFlag) => DONE
# - bool save(SfMData& sfmData, string& filename, ESfMData partFlag) => DONE
##################
def test_sfmdataio_load():
""" Test loading an SfMData file. """
data = sfmData.SfMData()
ret = sfmDataIO.load(data, SFMDATA_PATH, sfmDataIO.ALL)
assert ret, "Loading the SfMData file should have been successful as it is a valid one"
assert len(data.getViews()) == 30
def test_sfmdataio_save():
""" Test loading an SfMData file, editing it, and saving it. """
data = sfmData.SfMData()
ret = sfmDataIO.load(data, SFMDATA_PATH, sfmDataIO.ALL)
assert ret, "Loading the SfMData file should have been successful as it is a valid one"
views = data.getViews()
nb_views = len(views)
new_view = sfmData.View(IMAGE_PATH, VIEW_ID, INTRINSIC_ID, POSE_ID,
IMAGE_WIDTH, IMAGE_HEIGHT, RIG_ID, SUBPOSE_ID, METADATA)
views[VIEW_ID] = new_view
assert len(data.getViews()) == nb_views + 1
new_path = os.path.abspath(os.path.dirname(__file__)) + "/out.sfm"
ret = sfmDataIO.save(data, new_path, sfmDataIO.ALL)
assert ret
try:
new_data = sfmData.SfMData()
_ = sfmDataIO.load(new_data, new_path, sfmDataIO.ALL)
assert len(new_data.getViews()) == nb_views + 1
view = data.getView(VIEW_ID)
assert view == new_view
finally:
os.remove(new_path)
def test_sfmdataio_valid_ids():
""" Test loading an SfMData file and checking if it contains valid IDs. """
data = sfmData.SfMData()
ret = sfmDataIO.load(data, SFMDATA_PATH, sfmDataIO.ALL)
assert ret, "Loading the SfMData file should have been successful as it is a valid one"
assert sfmDataIO.validIds(data, sfmDataIO.ALL)
# Add a default View object at index 12345
views = data.getViews()
assert len(views) == 30
view = sfmData.View()
view.setViewId(12345)
views[12345] = view
assert len(data.getViews()) == 31
assert sfmDataIO.validIds(data, sfmDataIO.ALL)
# Set random intrinsic ID for the View that has been added
view.setIntrinsicId(23456)
assert not sfmDataIO.validIds(data, sfmDataIO.ALL)
|
