File tree Expand file tree Collapse file tree 1 file changed +7
-4
lines changed
gcm/tests/health_checks_tests Expand file tree Collapse file tree 1 file changed +7
-4
lines changed Original file line number Diff line number Diff line change 11# Copyright (c) Meta Platforms, Inc. and affiliates.
22# All rights reserved.
3+ import sys
34from unittest .mock import patch
45
56from gcm .health_checks .checks .check_nvidia_smi import NvidiaSmiCliImpl
89
910def test_nvidia_smi_cli_impl_uses_hal_adapter () -> None :
1011 # Patch default_backend_factories to avoid actual registry access
12+ # We use patch.object on the module to avoid ambiguity because
13+ # gcm.health_checks.checks.check_nvidia_smi resolves to the command function
14+ # when accessed via package attribute traversal in mock.patch string.
15+ module = sys .modules [NvidiaSmiCliImpl .__module__ ]
1116 with (
12- patch ("gcm.health_checks.checks.check_nvidia_smi.default_backend_factories" ),
13- patch (
14- "gcm.health_checks.checks.check_nvidia_smi.AcceleratorManager"
15- ) as MockManager ,
17+ patch .object (module , "default_backend_factories" ),
18+ patch .object (module , "AcceleratorManager" ) as MockManager ,
1619 ):
1720 # Mock manager instance
1821 manager_instance = MockManager .return_value
You can’t perform that action at this time.
0 commit comments