1 files changed, 114 insertions, 0 deletions
diff --git a/migration-test/tests/ApiTest.py b/migration-test/tests/ApiTest.py
new file mode 100644
index 0000000..0af8626
--- /dev/null
+++ b/migration-test/tests/ApiTest.py
@@ -0,0 +1,114 @@
+import csv
+import requests
+import random
+import os
+import unittest
+
+from apitest.pzbs import PzbsCalculator
+from apitest.api import ApiCalculator
+
+def runTest(tester, params):
+    return tester.get_response(
+        requests.post(
+            tester.get_url(),
+            data=tester.get_request(*params)
+        ).text)
+
+class ApiTestCase(unittest.TestCase):
+    def __setupParams(self):
+        csv_files = []
+        for root, dirs, files in os.walk('./csv'):
+            csv_files += [os.path.join(root, f) for f in files if f.lower().endswith('.csv')]
+        self.params = []
+        for csv_file in csv_files:
+            csv_content = csv.reader(open(csv_file), delimiter=";")
+            param_contents = [
+                row for idx, row in enumerate(csv_content) if idx == 2][0]
+            self.params.append((
+                param_contents[0][0].lower(),
+                None, 0,
+                int(param_contents[1]), float(param_contents[4])))
+
+    def setUp(self):
+        self.pzbs = PzbsCalculator()
+        self.api = ApiCalculator()
+        self.__setupParams()
+        self.longMessage=True
+        self.maxDiff = None
+
+    def test_csvCases(self):
+        for params in self.params:
+            for rank in ApiCalculator.ranks.keys():
+                for boards in [0, 1]:
+                    p = (params[0], rank, boards, params[3], params[4])
+                    self.assertEqual(
+                        runTest(self.pzbs, p),
+                        runTest(self.api, p),
+                        msg=str(p))
+
+    def test_randomCases(self):
+        for i in range(0, 20):
+            for type in ['t', 'p', 'i']:
+                for rank in ApiCalculator.ranks.keys():
+                    for boards in [0, 1]:
+                        p = (type, rank, boards,
+                             random.randint(6, 350),
+                             random.randint(0, 4000) / 2.0)
+                        self.assertEqual(
+                            runTest(self.pzbs, p),
+                            runTest(self.api, p),
+                            msg=str(p))
+
+    def test_customParamCases(self):
+        for type in ['t', 'p', 'i']:
+            for rank in ApiCalculator.ranks.keys():
+                for boards in [0, 1]:
+                    for j in range(0, 20):
+                        manualParams = {
+                            'points': random.randint(20, 100)
+                        }
+                        p = (type, rank, boards,
+                             35, 400, manualParams)
+                        self.assertEqual(
+                            runTest(self.pzbs, p),
+                            runTest(self.api, p),
+                            msg=str(p))
+                    for j in range(0, 20):
+                        manualParams = {
+                            'weight': random.randint(20, 100)
+                        }
+                        p = (type, rank, boards,
+                             35, 400, manualParams)
+                        self.assertEqual(
+                            runTest(self.pzbs, p),
+                            runTest(self.api, p),
+                            msg=str(p))
+                    for j in range(0, 20):
+                        manualParams = {
+                            'players': random.randint(20, 100) * 0.001
+                        }
+                        p = (type, rank, boards,
+                             35, 400, manualParams)
+                        self.assertEqual(
+                            runTest(self.pzbs, p),
+                            runTest(self.api, p),
+                            msg=str(p))
+                    for j in range(0, 20):
+                        manualParams = {
+                            'cutoff': [
+                                [random.uniform(0.04, 0.06),
+                                 random.uniform(0.7, 0.9)],
+                                [random.uniform(0.15, 0.35),
+                                 random.uniform(0.15, 0.35)],
+                                [random.uniform(0.4, 0.6),
+                                 0.0]
+                            ]
+                        }
+                        p = (type, rank, boards,
+                             random.randint(20, 25),
+                             random.randint(300, 400) / 2.0,
+                             manualParams)
+                        self.assertEqual(
+                            runTest(self.pzbs, p),
+                            runTest(self.api, p),
+                            msg=str(p))