#[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"),
    }
}