summaryrefslogtreecommitdiff
path: root/tests/vmtests/test_apt_config_cmd.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/vmtests/test_apt_config_cmd.py')
-rw-r--r--tests/vmtests/test_apt_config_cmd.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/tests/vmtests/test_apt_config_cmd.py b/tests/vmtests/test_apt_config_cmd.py
index 12962405..e957a1f2 100644
--- a/tests/vmtests/test_apt_config_cmd.py
+++ b/tests/vmtests/test_apt_config_cmd.py
@@ -5,10 +5,10 @@
apt-config standalone command.
"""
import textwrap
-import yaml
from . import VMBaseClass
from .releases import base_vm_classes as relbase
+from curtin.config import load_config
class TestAptConfigCMD(VMBaseClass):
@@ -50,7 +50,7 @@ class TestAptConfigCMD(VMBaseClass):
# For earlier than xenial 'apt_preserve_sources_list' is expected
self.assertEqual(
{'apt': {'preserve_sources_list': True}},
- yaml.load(self.load_collect_file("curtin-preserve-sources.cfg")))
+ load_config(self.collect_path("curtin-preserve-sources.cfg")))
class XenialTestAptConfigCMDCMD(relbase.xenial, TestAptConfigCMD):