import unittest import numpy as np class MyTestCase(unittest.TestCase): def test_randn(self): data = np.random.randn(2,3) print(data) if __name__ == '__main__': unittest.main()