Add unit test

.gitignore

@@ -4,4 +4,6 @@ __pycache__/
 mentors.csv
 teams.csv
 matching.csv
-compatibility.csv
+compatibility.csv
+.vscode/
+*.pyc

assign.py → mentor_matching/assign.py

@@ -12,7 +12,7 @@
 
 print("Process started!  Reading mentor file...", flush = True)
 mentors = []
-with open("mentors.csv") as mentorFile:
+with open("data/mentors.csv") as mentorFile:
 	mentorReader = csv.reader(mentorFile)
 	# remove header rows, if any
 	for _ in range(utils.mentorHeaderRows):
@@ -22,7 +22,7 @@
 
 print("Reading team file...", flush = True)
 teams = []
-with open("teams.csv") as teamFile:
+with open("data/teams.csv") as teamFile:
 	teamReader = csv.reader(teamFile)
 	# remove header rows, if any
 	for _ in range(utils.teamHeaderRows):

mentor_matching/mentor_test.py

@@ -0,0 +1,36 @@
+import csv
+import unittest
+
+from utils import Mentor
+
+
+class TestParsingMentor(unittest.TestCase):
+    def test_from_string(self):
+        raw_csv_line = [
+            "Mavericl,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,1,1,1,1,1,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,1,1,1,1,1,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,1,1,1,1,1,1,1,1,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,1,1,0,0,,,Jessica,1,Inconvenient,Convenient,Convenient,Not possible,Not possible,Somewhat,Very Confident"
+        ]
+        reader = csv.reader(raw_csv_line)
+        processed_line = next(reader)
+        mentor = Mentor(processed_line)
+
+        self.assertEqual(mentor.name, "Mavericl")
+        self.assertEqual(mentor.teamTypeRequests, [1, 1, 0, 0])
+        self.assertEqual(mentor.teamsRequested, [])
+        self.assertEqual(mentor.teamsRequired, [])
+        self.assertEqual(mentor.mentorsRequired, ["Jessica"])
+        self.assertEqual(mentor.comfortAlone, "1")
+        self.assertEqual(
+            mentor.transitConveniences,
+            [
+                "Inconvenient",
+                "Convenient",
+                "Convenient",
+                "Not possible",
+                "Not possible",
+            ],
+        )
+        self.assertEqual(mentor.skillsConfidence, ["Somewhat", "Very Confident"])
+
+
+if __name__ == "__main__":
+    unittest.main()