problems with mocking

test/test_helpers.py

@@ -5,6 +5,7 @@ import os
 import keras
 import numpy as np
 import mock
+import platform
 
 
 class TestToList:
@@ -168,7 +169,7 @@ class TestPrepareHost:
         with pytest.raises(OSError) as e:
             prepare_host()
         assert "unknown host 'NotExistingHostName'" in e.value.args[0]
-        if "runner-6HmDp9Qd-project-2411-concurrent" not in socket.gethostname():
+        if "runner-6HmDp9Qd-project-2411-concurrent" not in platform.node():
             mock_host.return_value = "linux-gzsx"
             with pytest.raises(NotADirectoryError) as e:
                 prepare_host()