diff --git a/.gitignore b/.gitignore
index 714dd72..8d64fcd 100644
--- a/.gitignore
+++ b/.gitignore
@@ -4,3 +4,4 @@
 /src/answer.py
 /html/
 /submission/5062CEM_2021_22_SepJan_CW1_main_sit_0123456789.html
+/tests/__pycache__/
diff --git a/requirements.txt b/requirements.txt
index 7678359..38ba1ba 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,4 @@
 simple-term-menu
 pdoc3
 markdown
-
+pytest
diff --git a/tests/test_solution.py b/tests/test_solution.py
new file mode 100644
index 0000000..a6efbf5
--- /dev/null
+++ b/tests/test_solution.py
@@ -0,0 +1,13 @@
+import pytest
+import sys
+sys.path.append("./src/")
+import rainbow_generator
+import pearson
+
+flag="MzZwcx15NHQzOnxzcnI9cjk="
+
+
+def test_generator():
+    d=rainbow_generator.generateTable(["test1", "test2", "test3"], lambda x: pearson.hashN(x,2), rainbow_generator.makeGuess, 20, 3, 6, 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789')
+    
+    assert d=={b'fa': 'test1', b'=\xcd': 'test2', b'\x0b\n': 'test3'}