def test_to_and_from_gpu(self):
""" Make sure we can load and unload data off the gpu. """
shape = (100,100,100)
d = Data()
space.initialize_space(shape)
for dtype in self.valid_dtypes:
# Create data to load.
d_cpu = np.random.randn(*shape).astype(dtype)
if dtype in (np.complex64, np.complex128):
d_cpu = (1 + 1j) * d_cpu
# Load and retrieve.
d.to_gpu(d_cpu)
self.assertTrue((d_cpu == d.get()).all())
