Create test.py

some silly test for CUDA