1 files changed, 65 insertions, 0 deletions
diff --git a/tests/fastrand_test.py b/tests/fastrand_test.py
new file mode 100755
index 0000000..d3bdbf1
--- /dev/null
+++ b/tests/fastrand_test.py
@@ -0,0 +1,65 @@
+#!/usr/bin/python -S
+#
+# Copyright 2014 Google Inc. All rights reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""
+fastrand_test.py: Tests for _fastrand extension module.
+"""
+import unittest
+
+import _fastrand  # module under test
+
+
+BIT_WIDTHS = [8, 16, 32, 64]
+
+
+class FastRandTest(unittest.TestCase):
+
+  def testRandbits64(self):
+    for n in BIT_WIDTHS:
+      #print '== %d' % n
+      for p1 in [0.1, 0.5, 0.9]:
+        #print '-- %f' % p1
+        for i in xrange(5):
+          r = _fastrand.randbits(p1, n)
+          # Rough sanity check
+          self.assertLess(r, 2 ** n)
+
+          # Visual check
+          #b = bin(r)
+          #print b
+          #print b.count('1')
+
+
+  def testRandbits64_EdgeCases(self):
+    for n in BIT_WIDTHS:
+      r = _fastrand.randbits(0.0, n)
+      self.assertEqual(0, r)
+
+    for n in BIT_WIDTHS:
+      r = _fastrand.randbits(1.0, n)
+      self.assertEqual(2 ** n - 1, r)
+
+  def testRandbitsError(self):
+    r = _fastrand.randbits(-1, 64)
+    # TODO: Should probably raise exceptions
+    self.assertEqual(None, r)
+
+    r = _fastrand.randbits(0.0, 65)
+    self.assertEqual(None, r)
+
+
+if __name__ == '__main__':
+  unittest.main()