these test should pass

waggle-sensor · Aug 16, 2024 · cad9f8e · cad9f8e
1 parent 548e84d
commit cad9f8e
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/app/test_calculate_pt.py b/app/test_calculate_pt.py
@@ -11,7 +11,13 @@ def test_case_1(self):
     def test_case_2(self):
         sdir = 18
         pdir = "wh,SEG,ES,neb,NG,NES,eg"
-        expected = '26, 16, 9, 7, 4, 29, 28'
+        expected = '26, 16, 9, 7, 4, 5, 12'
+        self.assertEqual(calculate_pt(sdir, pdir), expected)
+
+    def test_case_3(self):
+        sdir = 20
+        pdir = "SWS,EG,neh,NS,NWG,wb,eH,nEG"
+        expected = '21, 12, 6, 1, 32, 27, 10, 8'
         self.assertEqual(calculate_pt(sdir, pdir), expected)
 
     def test_invalid_direction(self):