summaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to 'src')
-rw-r--r--src/tests/helper.py1
-rw-r--r--src/tests/test_initial_setup.py1
2 files changed, 2 insertions, 0 deletions
diff --git a/src/tests/helper.py b/src/tests/helper.py
index 636e1ec..c3aed20 100644
--- a/src/tests/helper.py
+++ b/src/tests/helper.py
@@ -18,6 +18,7 @@ import sys
import importlib.util
import math
import json
+import os
def prepare_module(file_path='', module_name=''):
spec = importlib.util.spec_from_file_location(module_name, file_path)
diff --git a/src/tests/test_initial_setup.py b/src/tests/test_initial_setup.py
index f85bf12..4a0f371 100644
--- a/src/tests/test_initial_setup.py
+++ b/src/tests/test_initial_setup.py
@@ -17,6 +17,7 @@
import unittest
import vyos.configtree
import vyos.initialsetup as vis
+import json
from unittest import TestCase
from vyos.xml_ref import definition