Simplify tests

KarelZe · Dec 4, 2023 · 002e939 · 002e939
1 parent 4d78850
commit 002e939
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/tests/test_classical_classifier.py b/tests/test_classical_classifier.py
@@ -1,7 +1,4 @@
-"""Tests for the classical classifier.
-
-Use of artificial data to test the classifier.
-"""
+"""Tests for the classical classifier."""
 
 import numpy as np
 import pandas as pd
@@ -25,7 +22,7 @@ def setup(self) -> None:
         Prepares inputs and expected outputs for testing.
         """
         self.x_train = pd.DataFrame(
-            [[1, 2], [3, 4], [1, 2], [3, 4]], columns=["ask_best", "bid_best"]
+            [[1, 1], [1, 1], [1, 1], [1, 1]], columns=["ask_best", "bid_best"]
         )
         self.y_train = pd.Series([1, 1, -1, -1])
         self.x_test = pd.DataFrame(