Unit test for Sieve of Eratosthenes.

vedangmehta · vedangmehta · commit b61fa5772e8b · 2015-08-11T01:14:50.000+05:30
diff --git a/tests/sieve_test.py b/tests/sieve_test.py
@@ -0,0 +1,10 @@
+import unittest
+from sieve_of_eratosthenes import calculate_primes
+
+class TestSieveOfEratosthenes(unittest.TestCase):
+	def test_primes(self):
+		self.prime_list = [2,3,5,7,11,13,17,19]
+		self.assertEqual(self.prime_list,calculate_primes(20))
+
+if __name__ == '__main__':
+	unittest.main()
