Added unit tests for knuth_shuffle

nryoung · nryoung · commit 9cf075487392 · 2012-09-23T22:23:30.000-06:00
diff --git a/algorithms/tests/test_shuffling.py b/algorithms/tests/test_shuffling.py
@@ -0,0 +1,25 @@
+import unittest
+from ..shuffling import knuth
+
+class ShufflingAlgorithmTestCase(unittest.TestCase):
+    """
+    Shared code for shuffling unit tests.
+    """
+
+    def setUp(self):
+       self.sorted = range(10)
+
+
+class TestKnuthShuffle(ShufflingAlgorithmTestCase):
+    """
+    Tests Knuth shuffle on a small range from 0-9
+    """
+    def test_knuthshuffle(self):
+        self.shuffle = knuth.shuffle(range(10))
+
+        for i in self.shuffle:
+            not_shuffled = 0
+            if i == self.sorted[i]:
+                not_shuffled = not_shuffled + 1
+
+        self.assertGreater(2, not_shuffled)
diff --git a/algorithms/tests/test_sorting.py b/algorithms/tests/test_sorting.py
@@ -12,8 +12,6 @@ def setUp(self):
         self.input = range(10)
         random.shuffle(self.input)
         self.correct = range(10)
-        print self.input
-        print self.correct
 
 
 class TestBubbleSort(SortingAlgorithmTestCase):
