RexMex corrections.

Author: benedekrozemberczki

tests/integration/test_aggregation.py

@@ -17,7 +17,7 @@ def test_classification(self):
         assert performance_metrics.shape == (5, 8)
 
         performance_metrics = score_card.generate_report(scores, groupping=["source_group", "target_group"])
-        assert performance_metrics.shape == (25, 8)
+        assert performance_metrics.shape == (20, 8)
 
     def test_regression(self):
         reader = DatasetReader()
@@ -30,7 +30,7 @@ def test_regression(self):
         assert performance_metrics.shape == (5, 7)
 
         performance_metrics = score_card.generate_report(scores, groupping=["source_group", "target_group"])
-        assert performance_metrics.shape == (25, 7)
+        assert performance_metrics.shape == (20, 7)
 
     def test_addition(self):
         reader = DatasetReader()
@@ -42,4 +42,4 @@ def test_addition(self):
         assert performance_metrics.shape == (5, 15)
 
         performance_metrics = score_card.generate_report(scores, groupping=["source_group", "target_group"])
-        assert performance_metrics.shape == (25, 15)
+        assert performance_metrics.shape == (20, 15)

tests/unit/test_dataset.py

@@ -11,7 +11,7 @@ def test_erdos_renyi_structure(self):
         reader = DatasetReader()
         dataset = reader.read_dataset()
 
-        assert dataset.shape[0] == 50235
+        assert dataset.shape[0] == 50378
         assert dataset.shape[1] == 6
         assert set(["source_group", "target_group", "source_id", "target_id", "y_score", "y_true"]).issubset(dataset.columns) == True