Merge pull request #24 from vedangmehta/patch-5

Unit test for Sieve of Eratosthenes.
prakhar1989 · Aug 10, 2015 · 7444fcc · 7444fcc
2 parents e218d8e + b61fa57
commit 7444fcc
Showing 1 changed file with 10 additions and 0 deletions.
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()