From bf06d61c3572e7a6d41737a30e2f941a5f81b7a6 Mon Sep 17 00:00:00 2001 From: Harshil Patel Date: Tue, 5 Sep 2023 12:39:38 -0700 Subject: [PATCH] stdlib, tests, resources: Updated tests - Updated workload tests to use WrokloadResource and obtain_resource Change-Id: I39194e7fe764566a528e5141c29f30efa14e0cde --- tests/pyunit/stdlib/resources/pyunit_workload_checks.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/pyunit/stdlib/resources/pyunit_workload_checks.py b/tests/pyunit/stdlib/resources/pyunit_workload_checks.py index ef7c3caba1..a9fa5e28e2 100644 --- a/tests/pyunit/stdlib/resources/pyunit_workload_checks.py +++ b/tests/pyunit/stdlib/resources/pyunit_workload_checks.py @@ -32,6 +32,7 @@ from gem5.resources.resource import ( BinaryResource, DiskImageResource, obtain_resource, + WorkloadResource, ) from typing import Dict @@ -61,7 +62,7 @@ class CustomWorkloadTestSuite(unittest.TestCase): new=ClientWrapper(mock_config_json), ) def setUpClass(cls) -> None: - cls.custom_workload = CustomWorkload( + cls.custom_workload = WorkloadResource( function="set_se_binary_workload", parameters={ "binary": obtain_resource( @@ -135,7 +136,7 @@ class WorkloadTestSuite(unittest.TestCase): ClientWrapper(mock_config_json), ) def setUpClass(cls): - cls.workload = Workload("simple-boot", gem5_version="develop") + cls.workload = obtain_resource("simple-boot", gem5_version="develop") def test_get_function_str(self) -> None: # Tests `Resource.get_function_str`