diff --git a/optional_plugins/varianter_yaml_to_mux/tests/test_mux.py b/optional_plugins/varianter_yaml_to_mux/tests/test_mux.py index 3af317601da44baaeb52937cfcbc463da08408ae..87a6c5e190c1927681c684ec668fa09f6e0e52a8 100644 --- a/optional_plugins/varianter_yaml_to_mux/tests/test_mux.py +++ b/optional_plugins/varianter_yaml_to_mux/tests/test_mux.py @@ -2,9 +2,10 @@ import copy import itertools import os import pickle +import sys import unittest -import yaml +import yaml from six import iteritems import avocado_varianter_yaml_to_mux as yaml_to_mux @@ -420,8 +421,13 @@ class TestMultipleLoaders(unittest.TestCase): debug = yaml_to_mux.create_from_yaml([yaml_url], debug=True) self.assertEqual(type(debug), mux.MuxTreeNodeDebug) # Debug nodes are of generated "NamedTreeNodeDebug" type - self.assertEqual("", - str(type(debug.children[0]))) + if sys.version_info[0] == 3: + children_type = (".NamedTreeNodeDebug'>") + else: + children_type = ("") + self.assertEqual(children_type, str(type(debug.children[0]))) plain = yaml.load("foo: bar") self.assertEqual(type(plain), dict)