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

Fix be_print_strees_test

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2022-05-10 14:14:46 +02:00
parent e16dde713c
commit aebf301b29
1 changed files with 9 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
benchmark/tests/scripts/Be_Print_Strees_test.py
View File

@@ -10,39 +10,25 @@ class BePrintStrees(TestBase):
self.score = "accuracy"
self.files = []
self.datasets = ["balloons", "balance-scale"]
self.expected = {
"balloons": {
"color": "107a73552a9ad66dee34a825d613a792",
"gray": "d2a1746459298e9a625f8c96264a6841",
},
"balance-scale": {
"color": "0ba25cfe6a64d01187dfd3106b0cea7a",
"gray": "ce51aa4680e73af44b93ca717621d728",
},
}
def tearDown(self) -> None:
self.remove_files(self.files, ".")
return super().tearDown()
def hash_file(self, name):
file_name = os.path.join(Folders.img, f"{name}.png")
self.files.append(file_name)
self.assertTrue(os.path.exists(file_name))
with open(file_name, "rb") as f:
return hashlib.md5(f.read()).hexdigest(), file_name
def test_be_print_strees_dataset_bn(self):
for name in self.datasets:
stdout, _ = self.execute_script(
"be_print_strees",
["-d", name, "-q", "1"],
)
computed_hash, file_name = self.hash_file(f"stree_{name}")
file_name = os.path.join(Folders.img, f"stree_{name}.png")
self.files.append(file_name)
self.assertTrue(os.path.exists(file_name))
self.assertEqual(
stdout.getvalue(), f"File {file_name} generated\n"
)
self.assertEqual(computed_hash, self.expected[name]["gray"])
computed_size = os.path.getsize(file_name)
self.assertGreater(computed_size, 29000)
def test_be_print_strees_dataset_color(self):
for name in self.datasets:
@@ -50,8 +36,10 @@ class BePrintStrees(TestBase):
"be_print_strees",
["-d", name, "-q", "1", "-c", "1"],
)
computed_hash, file_name = self.hash_file(f"stree_{name}")
file_name = os.path.join(Folders.img, f"stree_{name}.png")
self.files.append(file_name)
self.assertEqual(
stdout.getvalue(), f"File {file_name} generated\n"
)
self.assertEqual(computed_hash, self.expected[name]["color"])
computed_size = os.path.getsize(file_name)
self.assertGreater(computed_size, 30000)