From 863be5ef756518f8d213266f195a4c06b403d5fd Mon Sep 17 00:00:00 2001 From: gaxler Date: Sat, 30 Jan 2021 20:09:52 -0800 Subject: [PATCH] style fixes --- src/preprocessing/categorical_encoder.rs | 20 ++++++++++---------- 1 file changed, 10 insertions(+), 10 deletions(-) diff --git a/src/preprocessing/categorical_encoder.rs b/src/preprocessing/categorical_encoder.rs index 22cd052..b05a344 100644 --- a/src/preprocessing/categorical_encoder.rs +++ b/src/preprocessing/categorical_encoder.rs @@ -262,9 +262,9 @@ mod tests { } #[test] fn test_fit() { - let (X, _) = build_fake_matrix(); + let (x, _) = build_fake_matrix(); let params = OneHotEncoderParams::from_cat_idx(&[1, 3]); - let oh_enc = OneHotEncoder::fit(&X, params).unwrap(); + let oh_enc = OneHotEncoder::fit(&x, params).unwrap(); assert_eq!(oh_enc.series_encoders.len(), 2); let num_cat: Vec = oh_enc @@ -277,17 +277,17 @@ mod tests { #[test] fn matrix_transform_test() { - let (X, expectedX) = build_fake_matrix(); + let (x, expected_x) = build_fake_matrix(); let params = OneHotEncoderParams::from_cat_idx(&[1, 3]); - let oh_enc = OneHotEncoder::fit(&X, params).unwrap(); - let nm = oh_enc.transform(&X).unwrap(); - assert_eq!(nm, expectedX); + let oh_enc = OneHotEncoder::fit(&x, params).unwrap(); + let nm = oh_enc.transform(&x).unwrap(); + assert_eq!(nm, expected_x); - let (X, expectedX) = build_cat_first_and_last(); + let (x, expected_x) = build_cat_first_and_last(); let params = OneHotEncoderParams::from_cat_idx(&[0, 2]); - let oh_enc = OneHotEncoder::fit(&X, params).unwrap(); - let nm = oh_enc.transform(&X).unwrap(); - assert_eq!(nm, expectedX); + let oh_enc = OneHotEncoder::fit(&x, params).unwrap(); + let nm = oh_enc.transform(&x).unwrap(); + assert_eq!(nm, expected_x); } #[test]