Doctorado-ML/benchmark
1
0
Fork 0
mirror of https://github.com/Doctorado-ML/benchmark.git synced 2025-08-18 00:45:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Complete be_grid tests

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2022-05-08 01:31:03 +02:00
parent 5c4d5cb99e
commit 4fe1e10488
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
benchmark/tests/scripts/Be_Grid_test.py
View File

@@ -35,10 +35,10 @@ class BeGridTest(TestBase):
def test_be_grid_(self): def test_be_grid_(self):
stdout, stderr = self.execute_script( stdout, stderr = self.execute_script(
"be_grid", "be_grid",
["-m", "SVC", "-s", "accuracy", "--n_folds", "2", "-q", "1"], ["-m", "SVC", "-s", "accuracy", "--n_folds", "2"],
) )
self.assertEqual(stderr.getvalue(), "") expected = "Perform grid search with SVC model\n"
self.assertEqual(stdout.getvalue(), "") self.assertTrue(stdout.getvalue().startswith(expected))
name = Files.grid_output("accuracy", "SVC") name = Files.grid_output("accuracy", "SVC")
file_name = os.path.join(Folders.results, name) file_name = os.path.join(Folders.results, name)
with open(file_name, "r") as f: with open(file_name, "r") as f: