Test Module I'll use pytest for test. CI doesn't support CUDA. Please don't write any CUDA supported code in test moduel.