IVGCVSW-5200 Add import enabled optimizer options to PyArmNN

Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: Ic7c288fd829d7f1f1ae2910c47fbccdd635def8f

1 files changed, 10 insertions, 0 deletions

diff --git a/python/pyarmnn/test/test_network.py b/python/pyarmnn/test/test_network.py
index 679e640374..c24b113cdb 100644
--- a/python/pyarmnn/test/test_network.py
+++ b/python/pyarmnn/test/test_network.py
@@ -11,18 +11,28 @@ def test_optimizer_options_default_values():
     assert opt.m_ReduceFp32ToFp16 == False
     assert opt.m_Debug == False
     assert opt.m_ReduceFp32ToBf16 == False
+    assert opt.m_ImportEnabled == False
 
 def test_optimizer_options_set_values1():
     opt = ann.OptimizerOptions(True, True)
     assert opt.m_ReduceFp32ToFp16 == True
     assert opt.m_Debug == True
     assert opt.m_ReduceFp32ToBf16 == False
+    assert opt.m_ImportEnabled == False
 
 def test_optimizer_options_set_values2():
     opt = ann.OptimizerOptions(False, False, True)
     assert opt.m_ReduceFp32ToFp16 == False
     assert opt.m_Debug == False
     assert opt.m_ReduceFp32ToBf16 == True
+    assert opt.m_ImportEnabled == False
+
+def test_optimizer_options_set_values3():
+    opt = ann.OptimizerOptions(False, False, True, True)
+    assert opt.m_ReduceFp32ToFp16 == False
+    assert opt.m_Debug == False
+    assert opt.m_ReduceFp32ToBf16 == True
+    assert opt.m_ImportEnabled == True
 
 @pytest.fixture(scope="function")
 def get_runtime(shared_data_folder, network_file):