Fix test issues

benchmark/scripts/be_report.py

@@ -1,11 +1,6 @@
 #!/usr/bin/env python
-import numpy as np
-from benchmark.Experiments import Datasets
-from benchmark.Results import Report, Excel, SQL, ReportBest
-from benchmark.Utils import (
-    Files,
-    TextColor,
-)
+from benchmark.Results import Report, Excel, SQL, ReportBest, ReportDatasets
+from benchmark.Utils import Files
 from benchmark.Arguments import Arguments
 
 
@@ -15,32 +10,6 @@ If no argument is set, displays the datasets and its characteristics
 """
 
 
-def default_report():
-    sets = Datasets()
-    color_line = TextColor.LINE1
-    print(color_line, end="")
-    print(f"{'Dataset':30s} Samp. Feat Cls Balance")
-    print("=" * 30 + " ===== ==== === " + "=" * 40)
-    for line in sets:
-        X, y = sets.load(line)
-        color_line = (
-            TextColor.LINE2
-            if color_line == TextColor.LINE1
-            else TextColor.LINE1
-        )
-        values, counts = np.unique(y, return_counts=True)
-        comp = ""
-        sep = ""
-        for value, count in zip(values, counts):
-            comp += f"{sep}{count/sum(counts)*100:5.2f}%"
-            sep = "/ "
-        print(color_line, end="")
-        print(
-            f"{line:30s} {X.shape[0]:5,d} {X.shape[1]:4d} "
-            f"{len(np.unique(y)):3d} {comp:40s}"
-        )
-
-
 def main():
     arguments = Arguments()
     arguments.xset("file").xset("excel").xset("sql").xset("compare")
@@ -48,11 +17,10 @@ def main():
         "score"
     )
     args = arguments.parse()
-
     if args.grid:
         args.best = False
     if args.file is None and args.best is None:
-        default_report()
+        ReportDatasets.report()
     else:
         if args.best is not None or args.grid is not None:
             report = ReportBest(args.score, args.model, args.best, args.grid)