def test_empty_node_snapshot(self): node = ParameterNode() snapshot = node.snapshot() self.assertCountEqual(snapshot.keys(), [ '__class__', 'functions', 'parameters', 'submodules', 'parameter_nodes' ]) node.name = 'node_name' self.assertEqual(node.snapshot()['name'], 'node_name')
def test_parameter_str(self): p = Parameter(set_cmd=None) self.assertEqual(str(p), 'None') p = Parameter('param1', set_cmd=None) self.assertEqual(str(p), 'param1') node = ParameterNode() p.parent = node self.assertEqual(str(p), 'param1') node.name = 'node1' self.assertEqual(str(p), 'node1_param1')