tests: Round predictions before equality assertion

Signed-off-by: John Andersen <johnandersenpdx@gmail.com>

Author: pdxjohnny

tests/integration/test_sources.py

@@ -104,4 +104,4 @@ async def test_string_keys(self):
         self.assertIn("Salary", results["prediction"])
         results = results["prediction"]["Salary"]
         self.assertIn("value", results)
-        self.assertEqual(70.0, results["value"])
+        self.assertEqual(70, round(results["value"]))

tests/test_high_level.py

@@ -130,8 +130,8 @@ async def test_predict(self):
         predictions = [
             prediction async for prediction in predict(model, predict_data)
         ]
-        self.assertEqual(predictions[0][2]["Salary"]["value"], 70)
-        self.assertEqual(predictions[1][2]["Salary"]["value"], 80)
+        self.assertEqual(round(predictions[0][2]["Salary"]["value"]), 70)
+        self.assertEqual(round(predictions[1][2]["Salary"]["value"]), 80)
 
 
 class TestDataFlow(TestOrchestrator):

tests/test_noasync.py

@@ -70,5 +70,5 @@ def test_predict(self):
         predictions = [
             prediction for prediction in predict(model, predict_data)
         ]
-        self.assertEqual(predictions[0][2]["Salary"]["value"], 70)
-        self.assertEqual(predictions[1][2]["Salary"]["value"], 80)
+        self.assertEqual(round(predictions[0][2]["Salary"]["value"]), 70)
+        self.assertEqual(round(predictions[1][2]["Salary"]["value"]), 80)