Skip to content

Commit

Permalink
Apply suggestions from code review
Browse files Browse the repository at this point in the history
compare data frames directly

Co-authored-by: Zhian N. Kamvar <[email protected]>
  • Loading branch information
elray1 and zkamvar authored Jan 7, 2025
1 parent 4afd00a commit 4cd6145
Showing 1 changed file with 2 additions and 30 deletions.
32 changes: 2 additions & 30 deletions tests/testthat/test-score_model_out_rel_metrics.R
Original file line number Diff line number Diff line change
Expand Up @@ -133,20 +133,7 @@ test_that("score_model_out succeeds with valid inputs: quantile output_type, rel
dplyr::full_join(exp_scores_relative_ae_median, by = c("model_id", "location")) |>
dplyr::full_join(exp_scores_relative_wis, by = c("model_id", "location"))

# same column names, number of rows, and score values
expect_equal(colnames(act_scores), colnames(exp_scores))
expect_equal(nrow(act_scores), nrow(exp_scores))
merged_scores <- dplyr::full_join(
act_scores, exp_scores,
by = c("model_id", "location")
)
expect_equal(nrow(act_scores), nrow(merged_scores))
expect_equal(merged_scores$ae_median.x, merged_scores$ae_median.y)
expect_equal(merged_scores$wis.x, merged_scores$wis.y)
expect_equal(merged_scores$interval_coverage_80.x, merged_scores$interval_coverage_80.y)
expect_equal(merged_scores$interval_coverage_90.x, merged_scores$interval_coverage_90.y)
expect_equal(merged_scores$ae_median_relative_skill.x, merged_scores$ae_median_relative_skill.y)
expect_equal(merged_scores$wis_relative_skill.x, merged_scores$wis_relative_skill.y)
expect_equal(act_scores, exp_scores, ignore_attr = TRUE)
})


Expand Down Expand Up @@ -234,22 +221,7 @@ test_that("score_model_out succeeds with valid inputs: quantile output_type, rel
dplyr::full_join(exp_scores_relative_ae_median, by = c("model_id", "location")) |>
dplyr::full_join(exp_scores_relative_wis, by = c("model_id", "location"))

# same column names, number of rows, and score values
expect_equal(colnames(act_scores), colnames(exp_scores))
expect_equal(nrow(act_scores), nrow(exp_scores))
merged_scores <- dplyr::full_join(
act_scores, exp_scores,
by = c("model_id", "location")
)
expect_equal(nrow(act_scores), nrow(merged_scores))
expect_equal(merged_scores$ae_median.x, merged_scores$ae_median.y)
expect_equal(merged_scores$wis.x, merged_scores$wis.y)
expect_equal(merged_scores$interval_coverage_80.x, merged_scores$interval_coverage_80.y)
expect_equal(merged_scores$interval_coverage_90.x, merged_scores$interval_coverage_90.y)
expect_equal(merged_scores$ae_median_relative_skill.x, merged_scores$ae_median_relative_skill.y)
expect_equal(merged_scores$ae_median_scaled_relative_skill.x, merged_scores$ae_median_scaled_relative_skill.y)
expect_equal(merged_scores$wis_relative_skill.x, merged_scores$wis_relative_skill.y)
expect_equal(merged_scores$wis_scaled_relative_skill.x, merged_scores$wis_scaled_relative_skill.y)
expect_equal(act_scores, exp_scores, ignore_attr = TRUE)
})


Expand Down

0 comments on commit 4cd6145

Please sign in to comment.