From f3a2886b45d5b2714dbe52cdd09ea0feebb8e126 Mon Sep 17 00:00:00 2001
From: Tom Offermann <tom@offermann.us>
Date: Mon, 4 Mar 2013 22:35:00 -0800
Subject: [PATCH] Add tests.

---
 test_randomSources.py | 56 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 56 insertions(+)
 create mode 100644 test_randomSources.py

diff --git a/test_randomSources.py b/test_randomSources.py
new file mode 100644
index 0000000..fe270a5
--- /dev/null
+++ b/test_randomSources.py
@@ -0,0 +1,56 @@
+import unittest
+
+import randomSources
+
+class TestQuantumRandom(unittest.TestCase):
+
+    def setUp(self):
+        self.rng = randomSources.QuantumRandom()
+
+    def test_quantum_randrange(self):
+        self.assertTrue(self.rng.randrange(100) in range(0, 100))
+
+    def test_quantum_randrange_start_stop(self):
+        self.assertTrue(self.rng.randrange(10, 100) in range(10, 100))
+
+    def test_quantum_choice(self):
+        self.assertTrue(self.rng.choice(['a', 'b', 'c']) in ['a', 'b', 'c'])
+
+    def test_quantum_shuffle(self):
+        orig = list('A23456789JQK')
+        cards = list('A23456789JQK')
+        self.rng.shuffle(cards)
+        self.assertEqual(set(cards), set(orig))
+
+    def test_quantum_sample(self):
+        cards = list('A23456789JQK')
+        self.assertTrue(set(self.rng.sample(cards, 5)).issubset(set(cards)))
+
+
+class TestRandomDotOrg(unittest.TestCase):
+
+    def setUp(self):
+        self.rng = randomSources.RandomDotOrg()
+
+    def test_randomdotorg_randrange(self):
+        self.assertTrue(self.rng.randrange(100) in range(100))
+
+    def test_randomdotorg_randrange_start_stop(self):
+        self.assertTrue(self.rng.randrange(10, 100) in range(10, 100))
+
+    def test_randomdotorg_choice(self):
+        self.assertTrue(self.rng.choice(['a', 'b', 'c']) in ['a', 'b', 'c'])
+
+    def test_randomdotorg_shuffle(self):
+        orig = list('A23456789JQK')
+        cards = list('A23456789JQK')
+        self.rng.shuffle(cards)
+        self.assertEqual(set(cards), set(orig))
+
+    def test_randomdotorg_sample(self):
+        cards = list('A23456789JQK')
+        self.assertTrue(set(self.rng.sample(cards, 5)).issubset(set(cards)))
+
+
+if __name__ == '__main__':
+    unittest.main()