@@ -114,16 +114,20 @@ def verify_vega(
114114 assert j [0 ]["type" ] == "vega"
115115 assert set (j [0 ]["revisions" ]) == set (versions )
116116
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>" ]
119120 )
121+
120122 assert set (versions ) == set (json_result [0 ]["revisions" ])
121123
122124 assert json_result [0 ]["content" ]["data" ]["values" ]
123125 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
127131
128132 def _assert_templates_equal (
129133 html_template , filled_template , split_template , title , x_label , y_label
@@ -133,7 +137,7 @@ def _assert_templates_equal(
133137 tmp1 = deepcopy (html_template )
134138 tmp2 = deepcopy (filled_template )
135139 tmp3 = json .loads (
136- split_template [:]
140+ json . dumps ( split_template )
137141 .replace ('"<DVC_METRIC_COLUMN_WIDTH>"' , "300" )
138142 .replace ('"<DVC_METRIC_PLOT_HEIGHT>"' , "300" )
139143 .replace ('"<DVC_METRIC_PLOT_WIDTH>"' , "300" )
@@ -486,11 +490,9 @@ def test_repo_with_config_plots(tmp_dir, capsys, repo_with_config_plots):
486490
487491 assert html_result ["linear_train_vs_test" ]["data" ]["values" ] == ble
488492 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+ ]
494496 == ble
495497 )
496498
@@ -501,7 +503,7 @@ def test_repo_with_dvclive_plots(tmp_dir, capsys, repo_with_dvclive_plots):
501503
502504 for s in ("show" , "diff" ):
503505 _ , json_result , split_json_result = call (capsys , subcommand = s )
504- expected_result = {
506+ expected_result : Dict [ str , Dict [ str , list [ str ]]] = {
505507 "data" : {
506508 "dvclive/plots/metrics/metric.tsv" : [],
507509 },
0 commit comments