@@ -114,16 +114,20 @@ def verify_vega(
114
114
assert j [0 ]["type" ] == "vega"
115
115
assert set (j [0 ]["revisions" ]) == set (versions )
116
116
117
- assert json_result [0 ]["content" ]["data" ]["values" ] == json .loads (
118
- split_json_result [0 ]["anchor_definitions" ]["<DVC_METRIC_DATA>" ]
117
+ assert (
118
+ json_result [0 ]["content" ]["data" ]["values" ]
119
+ == split_json_result [0 ]["anchor_definitions" ]["<DVC_METRIC_DATA>" ]
119
120
)
121
+
120
122
assert set (versions ) == set (json_result [0 ]["revisions" ])
121
123
122
124
assert json_result [0 ]["content" ]["data" ]["values" ]
123
125
assert html_result ["data" ]["values" ]
124
- assert "<DVC_METRIC_DATA>" in split_json_result [0 ]["content" ]
125
- assert "<DVC_METRIC_X_LABEL>" in split_json_result [0 ]["content" ]
126
- assert "<DVC_METRIC_Y_LABEL>" in split_json_result [0 ]["content" ]
126
+
127
+ content_str = json .dumps (split_json_result [0 ]["content" ])
128
+ assert "<DVC_METRIC_DATA>" in content_str
129
+ assert "<DVC_METRIC_X_LABEL>" in content_str
130
+ assert "<DVC_METRIC_Y_LABEL>" in content_str
127
131
128
132
def _assert_templates_equal (
129
133
html_template , filled_template , split_template , title , x_label , y_label
@@ -133,7 +137,7 @@ def _assert_templates_equal(
133
137
tmp1 = deepcopy (html_template )
134
138
tmp2 = deepcopy (filled_template )
135
139
tmp3 = json .loads (
136
- split_template [:]
140
+ json . dumps ( split_template )
137
141
.replace ('"<DVC_METRIC_COLUMN_WIDTH>"' , "300" )
138
142
.replace ('"<DVC_METRIC_PLOT_HEIGHT>"' , "300" )
139
143
.replace ('"<DVC_METRIC_PLOT_WIDTH>"' , "300" )
@@ -486,11 +490,9 @@ def test_repo_with_config_plots(tmp_dir, capsys, repo_with_config_plots):
486
490
487
491
assert html_result ["linear_train_vs_test" ]["data" ]["values" ] == ble
488
492
assert (
489
- json .loads (
490
- split_json_result ["data" ]["linear_train_vs_test" ][0 ]["anchor_definitions" ][
491
- "<DVC_METRIC_DATA>"
492
- ]
493
- )
493
+ split_json_result ["data" ]["linear_train_vs_test" ][0 ]["anchor_definitions" ][
494
+ "<DVC_METRIC_DATA>"
495
+ ]
494
496
== ble
495
497
)
496
498
@@ -501,7 +503,7 @@ def test_repo_with_dvclive_plots(tmp_dir, capsys, repo_with_dvclive_plots):
501
503
502
504
for s in ("show" , "diff" ):
503
505
_ , json_result , split_json_result = call (capsys , subcommand = s )
504
- expected_result = {
506
+ expected_result : Dict [ str , Dict [ str , list [ str ]]] = {
505
507
"data" : {
506
508
"dvclive/plots/metrics/metric.tsv" : [],
507
509
},
0 commit comments