diff options
Diffstat (limited to 'python/pyarmnn/test/test_runtime.py')
-rw-r--r-- | python/pyarmnn/test/test_runtime.py | 9 |
1 files changed, 3 insertions, 6 deletions
diff --git a/python/pyarmnn/test/test_runtime.py b/python/pyarmnn/test/test_runtime.py index e558e84e28..a37772c5df 100644 --- a/python/pyarmnn/test/test_runtime.py +++ b/python/pyarmnn/test/test_runtime.py @@ -156,7 +156,9 @@ def test_load_network_properties_provided(random_runtime): opt_network, _ = ann.Optimize(network, preferred_backends, runtime.GetDeviceSpec(), ann.OptimizerOptions()) - properties = ann.INetworkProperties(True, True) + inputSource = ann.MemorySource_Malloc + outputSource = ann.MemorySource_Malloc + properties = ann.INetworkProperties(False, inputSource, outputSource) net_id, messages = runtime.LoadNetwork(opt_network, properties) assert "" == messages assert net_id == 0 @@ -184,11 +186,6 @@ def test_network_properties_constructor(random_runtime): assert net_id == 0 -def test_network_properties_deprecated_constructor(): - with pytest.warns(DeprecationWarning): - warnings.warn("Deprecated: Use constructor with MemorySource argument instead.", DeprecationWarning) - - def test_unload_network_fails_for_invalid_net_id(random_runtime): preferred_backends = random_runtime[0] network = random_runtime[1] |