48 lines
1.3 KiB
Rust
48 lines
1.3 KiB
Rust
use rtx_preprocessing::error::PreprocessingError;
|
|
|
|
#[test]
|
|
fn test_invalid_shape_error() {
|
|
let err = PreprocessingError::InvalidShape {
|
|
message: "Shape [2, 3] is invalid for this operation".to_string(),
|
|
};
|
|
|
|
let err_msg = err.to_string();
|
|
assert!(err_msg.contains("Invalid shape"));
|
|
assert!(err_msg.contains("[2, 3]"));
|
|
}
|
|
|
|
#[test]
|
|
fn test_shape_mismatch_error() {
|
|
let err = PreprocessingError::ShapeMismatch {
|
|
expected: vec![10, 20],
|
|
actual: vec![10, 30],
|
|
};
|
|
|
|
let err_msg = err.to_string();
|
|
assert!(err_msg.contains("Shape mismatch"));
|
|
assert!(err_msg.contains("[10, 20]"));
|
|
assert!(err_msg.contains("[10, 30]"));
|
|
}
|
|
|
|
#[test]
|
|
fn test_dimension_mismatch_error() {
|
|
let err = PreprocessingError::DimensionMismatch {
|
|
expected: 3,
|
|
actual: 2,
|
|
};
|
|
|
|
let err_msg = err.to_string();
|
|
assert!(err_msg.contains("Dimension mismatch"));
|
|
assert!(err_msg.contains("3"));
|
|
assert!(err_msg.contains("2"));
|
|
}
|
|
|
|
#[test]
|
|
fn test_error_conversion_from_tensor() {
|
|
// Test that TensorError can be converted to PreprocessingError
|
|
let tensor_err = rtx_tensor::TensorError::shape("Shape mismatch: expected [1, 2], got [2, 1]");
|
|
|
|
let preprocessing_err: PreprocessingError = tensor_err.into();
|
|
assert!(matches!(preprocessing_err, PreprocessingError::Tensor(_)));
|
|
}
|