diff options
Diffstat (limited to 'test')
-rw-r--r-- | test/__init__.py | 0 | ||||
-rw-r--r-- | test/test_rand.py | 31 |
2 files changed, 31 insertions, 0 deletions
diff --git a/test/__init__.py b/test/__init__.py new file mode 100644 index 0000000..e69de29 --- /dev/null +++ b/test/__init__.py diff --git a/test/test_rand.py b/test/test_rand.py new file mode 100644 index 0000000..76b4d24 --- /dev/null +++ b/test/test_rand.py @@ -0,0 +1,31 @@ +import sys +import unittest + +import rand + + +class RandTestCase(unittest.TestCase): + def test_rand_status(self): + self.assertEqual(rand.rand_status(), 1) + + def test_rand_event(self): + if sys.platform == 'win32': + self.assertIsNotNone(rand.rand_event(1, 1, 1)) + else: + self.assertIsNone(rand.rand_event(1, 1, 1)) + + def test_rand_bytes(self): + exp_length = 15 + r = rand.rand_bytes(exp_length) + self.assertIsInstance(r, bytes) + self.assertEqual(len(r), exp_length) + + def test_rand_pseudo_bytes(self): + exp_length = 15 + r = rand.rand_pseudo_bytes(exp_length) + self.assertIsInstance(r[0], bytes) + self.assertEqual(len(r[0]), exp_length) + self.assertEqual(r[1], 1) + +if __name__ == '__main__': + unittest.main() |