add test for knapsack

UST-QuAntiL · Feb 6, 2024 · 0bb3bee · 0bb3bee
1 parent 9f0d4bb
commit 0bb3bee
Showing 1 changed file with 36 additions and 0 deletions.
diff --git a/tests/test_costFunctions.py b/tests/test_costFunctions.py
@@ -195,3 +195,39 @@ def test_max_cut(self):
         )
         self.assertEqual(response.status_code, 200)
         print(response.get_json())
+
+    def test_knapsack(self):
+        response = self.client.post(
+            "/objective/knapsack",
+            data=json.dumps(
+                {
+                    "items": [
+                        {
+                            "value": 5,
+                            "weight": 2
+                        },
+                        {
+                            "value": 2,
+                            "weight": 1
+                        },
+                        {
+                            "value": 3,
+                            "weight": 2
+                        }
+                    ],
+                    "max_weights": 20,
+                    "counts": {
+                        "100000": 30,
+                        "100001": 10,
+                        "110000": 50,
+                        "011110": 20,
+                        "010110": 40
+                    },
+                    "objFun": "Expectation",
+                    "visualization": "True"
+                }
+            ),
+            content_type="application/json",
+        )
+        self.assertEqual(response.status_code, 200)
+        print(response.get_json())