Adding more tests

Author: SaraTahir28

laptopAllocation/tests/test_laptop.py

@@ -1,7 +1,4 @@
-# test_basic.py
-# A very small test just to confirm the core logic works.
-# Full testing is beyond the scope of this assignment, but this shows
-# that the sadness function behaves correctly for a simple case.
+
 #If a person gets a laptop with their first‑choice operating system, their sadness should be 0.
 from laptop import Person, Laptop, OperatingSystem, sadness
 
@@ -10,3 +7,25 @@ def test_sadness_first_choice():
     laptop = Laptop(1, "Apple", "macBook", 13, OperatingSystem.MACOS)
 
     assert sadness(person, laptop) == 0
+from laptop import Person, Laptop, OperatingSystem, sadness, TERRIBLE_SADNESS
+#Adding more tests 
+def test_sadness_second_choice():
+    person = Person("Test", 20, (OperatingSystem.MACOS, OperatingSystem.UBUNTU))
+    laptop = Laptop(1, "Dell", "XPS", 13, OperatingSystem.UBUNTU)
+    assert sadness(person, laptop) == 1
+
+
+def test_sadness_not_in_preferences():
+    person = Person("Test", 20, (OperatingSystem.MACOS,))
+    laptop = Laptop(1, "Dell", "XPS", 13, OperatingSystem.UBUNTU)
+    assert sadness(person, laptop) == TERRIBLE_SADNESS
+
+
+def test_sadness_above_max_rank():
+    # as our MAX_HAPPY_RANK = 3, so rank 3 should be TERRIBLE_SADNESS
+    person = Person(
+        "Test", 20,
+        (OperatingSystem.MACOS, OperatingSystem.UBUNTU, OperatingSystem.ARCH)
+    )
+    laptop = Laptop(1, "Dell", "XPS", 13, OperatingSystem.ARCH)
+    assert sadness(person, laptop) == TERRIBLE_SADNESS