Clean up PcepOperations

[integration/test.git] / csit / libraries / Topology.py

diff --git a/csit/libraries/Topology.py b/csit/libraries/Topology.py
index 11a5c71fecc1e7252e534654337afabea56885fb..b82ff1d1dffba5bbd890d6f98db57d98862925a0 100644 (file)
--- a/csit/libraries/Topology.py
+++ b/csit/libraries/Topology.py
@@ -7,24 +7,27 @@ from robot.libraries.BuiltIn import BuiltIn
 
 
 class Topology(object):
-    '''
+    """
     Topology class provide topology database and provide many method to get property of topology.
-    '''
+    """
+
     topo_nodes_db = [
         [],
-        [{u'type': u'OF', u'id': u'00:00:00:00:00:00:00:01'}],
-        [{u'type': u'OF', u'id': u'00:00:00:00:00:00:00:01'},
-         {u'type': u'OF', u'id': u'00:00:00:00:00:00:00:02'},
-         {u'type': u'OF', u'id': u'00:00:00:00:00:00:00:03'}]
-        ]
+        [{"type": "OF", "id": "00:00:00:00:00:00:00:01"}],
+        [
+            {"type": "OF", "id": "00:00:00:00:00:00:00:01"},
+            {"type": "OF", "id": "00:00:00:00:00:00:00:02"},
+            {"type": "OF", "id": "00:00:00:00:00:00:00:03"},
+        ],
+    ]
 
     def __init__(self):
         self.builtin = BuiltIn()
 
     def get_nodes_from_topology(self, topo_level):
-        '''
+        """
         get nodes from topology database by topology tree level
-        '''
+        """
         if isinstance(topo_level, str) or isinstance(topo_level, unicode):
             if topo_level.isdigit():
                 topo_level = int(topo_level)
@@ -40,7 +43,8 @@ class Topology(object):
         else:
             return None
 
-if __name__ == '__main__':
+
+if __name__ == "__main__":
     topology = Topology()
-    print topology.get_nodes_from_topology(2)
-    print topology.get_nodes_from_topology('2')
+    print(topology.get_nodes_from_topology(2))
+    print(topology.get_nodes_from_topology("2"))