Skip to content

Commit f4f6b53

Browse files
committed
fix unit tests
1 parent 3ac97a1 commit f4f6b53

File tree

1 file changed

+26
-8
lines changed

1 file changed

+26
-8
lines changed

tests/charts/test_plotdefinition_manager.py

Lines changed: 26 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -35,10 +35,16 @@
3535
def spec():
3636
transfers = [
3737
DataTransferSpec(
38-
display_name="trans1", show_convergence=True, show_transfer_values=True
38+
name="TRANS-1",
39+
display_name="trans1",
40+
show_convergence=True,
41+
show_transfer_values=True,
3942
),
4043
DataTransferSpec(
41-
display_name="trans2", show_convergence=True, show_transfer_values=True
44+
name="TRANS-2",
45+
display_name="trans2",
46+
show_convergence=True,
47+
show_transfer_values=True,
4248
),
4349
]
4450

@@ -50,18 +56,30 @@ def spec():
5056
def spec2():
5157
transfers1 = [
5258
DataTransferSpec(
53-
display_name="trans1", show_convergence=True, show_transfer_values=True
59+
name="TRANS-1",
60+
display_name="trans1",
61+
show_convergence=True,
62+
show_transfer_values=True,
5463
),
5564
DataTransferSpec(
56-
display_name="trans2", show_convergence=True, show_transfer_values=True
65+
name="TRANS-2",
66+
display_name="trans2",
67+
show_convergence=True,
68+
show_transfer_values=True,
5769
),
5870
]
5971
transfers2 = [
6072
DataTransferSpec(
61-
display_name="Trans1", show_convergence=True, show_transfer_values=False
73+
name="TRANS-1",
74+
display_name="Trans1",
75+
show_convergence=True,
76+
show_transfer_values=False,
6277
),
6378
DataTransferSpec(
64-
display_name="Trans2", show_convergence=False, show_transfer_values=True
79+
name="TRANS-2",
80+
display_name="Trans2",
81+
show_convergence=False,
82+
show_transfer_values=True,
6583
),
6684
]
6785

@@ -134,7 +152,7 @@ def test_init_from_spec(spec):
134152

135153

136154
def test_set_metadata(spec, metadata):
137-
pdm = PlotDefinitionManager(spec)
155+
pdm = PlotDefinitionManager(spec, is_csv_source=True)
138156
mgr = pdm.subplot_mgr(metadata.name)
139157
mgr.set_metadata(metadata)
140158

@@ -149,7 +167,7 @@ def test_set_metadata(spec, metadata):
149167

150168

151169
def test_set_metadata_two_interfaces(spec2, metadata, metadata2):
152-
pdm = PlotDefinitionManager(spec2)
170+
pdm = PlotDefinitionManager(spec2, is_csv_source=True)
153171
mgr1 = pdm.subplot_mgr(metadata.name)
154172
mgr2 = pdm.subplot_mgr(metadata2.name)
155173

0 commit comments

Comments
 (0)