test/tensor: add specific test for sync/access API (autumnai#37)

alexandermorozov · Apr 17, 2016 · ce927ef · ce927ef
1 parent 728603e
commit ce927ef
Show file tree

Hide file tree

Showing 5 changed files with 23 additions and 4 deletions.
diff --git a/src/device.rs b/src/device.rs
@@ -64,7 +64,7 @@ pub enum DeviceType {
     Cuda(CudaContext),
 }
 
-#[derive(Debug, Copy, Clone)]
+#[derive(Debug, Copy, Clone, PartialEq, Eq)]
 /// Defines a generic set of Memory Errors.
 pub enum Error {
     /// Failures related to the Native framework implementation.

diff --git a/src/frameworks/cuda/api/driver/error.rs b/src/frameworks/cuda/api/driver/error.rs
@@ -2,7 +2,7 @@
 
 use std::{fmt, error};
 
-#[derive(Debug, Copy, Clone)]
+#[derive(Debug, Copy, Clone, PartialEq, Eq)]
 /// Defines OpenCL errors.
 pub enum Error {
     /// Failure with provided value.

diff --git a/src/frameworks/native/error.rs b/src/frameworks/native/error.rs
@@ -2,7 +2,7 @@
 
 use std::{fmt, error};
 
-#[derive(Debug, Copy, Clone)]
+#[derive(Debug, Copy, Clone, PartialEq, Eq)]
 /// Defines the Native Error.
 pub enum Error {
     /// Failure related to allocation, syncing memory

diff --git a/src/frameworks/opencl/api/error.rs b/src/frameworks/opencl/api/error.rs
@@ -2,7 +2,7 @@
 
 use std::{fmt, error};
 
-#[derive(Debug, Copy, Clone)]
+#[derive(Debug, Copy, Clone, PartialEq, Eq)]
 /// Defines OpenCL errors.
 pub enum Error {
     /// Failure with provided platform.

diff --git a/tests/shared_memory_specs.rs b/tests/shared_memory_specs.rs
@@ -4,6 +4,7 @@ extern crate libc;
 #[cfg(test)]
 mod shared_memory_spec {
     use co::prelude::*;
+    use co::tensor::Error;
 
     fn write_to_memory<T: Copy>(mem: &mut MemoryType, data: &[T]) {
         match mem {
@@ -59,6 +60,24 @@ mod shared_memory_spec {
         }
     }
 
+    #[test]
+    #[cfg(feature = "native")]
+    fn it_fails_on_initialized_memory_read() {
+        let ntv = Native::new();
+        let cpu = ntv.new_device(ntv.hardwares()).unwrap();
+        let shared_data = &mut SharedTensor::<f32>::new(&10).unwrap();
+        assert_eq!(shared_data.read(&cpu).unwrap_err(),
+                   Error::UninitializedMemory);
+        assert_eq!(shared_data.read_write(&cpu).unwrap_err(),
+                   Error::UninitializedMemory);
+
+        shared_data.write_only(&cpu).unwrap();
+        shared_data.drop_device(&cpu).unwrap();
+
+        assert_eq!(shared_data.read(&cpu).unwrap_err(),
+                   Error::UninitializedMemory);
+    }
+
     #[test]
     #[cfg(feature = "cuda")]
     fn it_syncs_from_native_to_cuda_and_back() {