Skip to content

Commit 9433a39

Browse files
committed
test: panic test for invalid pdf argument
1 parent 280fa7e commit 9433a39

2 files changed

Lines changed: 14 additions & 0 deletions

File tree

src/distribution/multivariate_normal.rs

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -373,4 +373,11 @@ mod tests {
373373
test_case(vec![0., 0.], vec![f64::INFINITY, 0., 0., f64::INFINITY], f64::NEG_INFINITY, ln_pdf(dvec![10., 10.]));
374374
test_case(vec![0., 0.], vec![f64::INFINITY, 0., 0., f64::INFINITY], f64::NEG_INFINITY, ln_pdf(dvec![100., 100.]));
375375
}
376+
377+
#[test]
378+
#[should_panic]
379+
fn test_pdf_mismatched_arg_size() {
380+
let mvn = MultivariateNormal::new(vec![0., 0.], vec![1., 0., 0., 1.,]).unwrap();
381+
mvn.pdf(&dvec![1.]); // x.size != mu.size
382+
}
376383
}

src/distribution/multivariate_students_t.rs

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -449,4 +449,11 @@ mod tests {
449449
test_almost_multivariate_normal(vec![0., 0.,], vec![1., 0., 0., 1.], 1e10, 1e-5, dvec![1., 1.], pdf_mvs, pdf_mvn);
450450
test_almost_multivariate_normal(vec![0., 0.,], vec![1., 0., 0., 1.], f64::INFINITY, 1e-50, dvec![1., 1.], pdf_mvs, pdf_mvn);
451451
}
452+
453+
#[test]
454+
#[should_panic]
455+
fn test_pdf_mismatched_arg_size() {
456+
let mvs = MultivariateStudent::new(vec![0., 0.], vec![1., 0., 0., 1.,], 3.).unwrap();
457+
mvs.pdf(&dvec![1.]); // x.size != mu.size
458+
}
452459
}

0 commit comments

Comments
 (0)