18 lines
377 B
Rust
18 lines
377 B
Rust
#[test]
|
|
fn test_library_compiles() {
|
|
// Basic test to ensure the library compiles
|
|
assert!(true);
|
|
}
|
|
|
|
#[test]
|
|
fn test_imports_work() {
|
|
// Test that we can use rtx_tensor types
|
|
use rtx_tensor::{Device, Shape};
|
|
|
|
let device = Device::Cuda(0);
|
|
match device {
|
|
Device::Cuda(id) => assert_eq!(id, 0),
|
|
_ => panic!("Expected CUDA device"),
|
|
}
|
|
}
|