1. import copy

  2. import unittest

  3. 

  4. from django.utils.tree import Node

  5. 

  6. 

  7. class NodeTests(unittest.TestCase):

  8.     def setUp(self):

  9.         self.node1_children = [("a", 1), ("b", 2)]

  10.         self.node1 = Node(self.node1_children)

  11.         self.node2 = Node()

  12. 

  13.     def test_str(self):

  14.         self.assertEqual(str(self.node1), "(DEFAULT: ('a', 1), ('b', 2))")

  15.         self.assertEqual(str(self.node2), "(DEFAULT: )")

  16. 

  17.     def test_repr(self):

  18.         self.assertEqual(repr(self.node1), "<Node: (DEFAULT: ('a', 1), ('b', 2))>")

  19.         self.assertEqual(repr(self.node2), "<Node: (DEFAULT: )>")

  20. 

  21.     def test_hash(self):

  22.         node3 = Node(self.node1_children, negated=True)

  23.         node4 = Node(self.node1_children, connector="OTHER")

  24.         node5 = Node(self.node1_children)

  25.         node6 = Node([["a", 1], ["b", 2]])

  26.         node7 = Node([("a", [1, 2])])

  27.         node8 = Node([("a", (1, 2))])

  28.         self.assertNotEqual(hash(self.node1), hash(self.node2))

  29.         self.assertNotEqual(hash(self.node1), hash(node3))

  30.         self.assertNotEqual(hash(self.node1), hash(node4))

  31.         self.assertEqual(hash(self.node1), hash(node5))

  32.         self.assertEqual(hash(self.node1), hash(node6))

  33.         self.assertEqual(hash(self.node2), hash(Node()))

  34.         self.assertEqual(hash(node7), hash(node8))

  35. 

  36.     def test_len(self):

  37.         self.assertEqual(len(self.node1), 2)

  38.         self.assertEqual(len(self.node2), 0)

  39. 

  40.     def test_bool(self):

  41.         self.assertTrue(self.node1)

  42.         self.assertFalse(self.node2)

  43. 

  44.     def test_contains(self):

  45.         self.assertIn(("a", 1), self.node1)

  46.         self.assertNotIn(("a", 1), self.node2)

  47. 

  48.     def test_add(self):

  49.         # start with the same children of node1 then add an item

  50.         node3 = Node(self.node1_children)

  51.         node3_added_child = ("c", 3)

  52.         # add() returns the added data

  53.         self.assertEqual(node3.add(node3_added_child, Node.default), node3_added_child)

  54.         # we added exactly one item, len() should reflect that

  55.         self.assertEqual(len(self.node1) + 1, len(node3))

  56.         self.assertEqual(str(node3), "(DEFAULT: ('a', 1), ('b', 2), ('c', 3))")

  57. 

  58.     def test_add_eq_child_mixed_connector(self):

  59.         node = Node(["a", "b"], "OR")

  60.         self.assertEqual(node.add("a", "AND"), "a")

  61.         self.assertEqual(node, Node([Node(["a", "b"], "OR"), "a"], "AND"))

  62. 

  63.     def test_negate(self):

  64.         # negated is False by default

  65.         self.assertFalse(self.node1.negated)

  66.         self.node1.negate()

  67.         self.assertTrue(self.node1.negated)

  68.         self.node1.negate()

  69.         self.assertFalse(self.node1.negated)

  70. 

  71.     def test_deepcopy(self):

  72.         node4 = copy.copy(self.node1)

  73.         node5 = copy.deepcopy(self.node1)

  74.         self.assertIs(self.node1.children, node4.children)

  75.         self.assertIsNot(self.node1.children, node5.children)

  76. 

  77.     def test_eq_children(self):

  78.         node = Node(self.node1_children)

  79.         self.assertEqual(node, self.node1)

  80.         self.assertNotEqual(node, self.node2)

  81. 

  82.     def test_eq_connector(self):

  83.         new_node = Node(connector="NEW")

  84.         default_node = Node(connector="DEFAULT")

  85.         self.assertEqual(default_node, self.node2)

  86.         self.assertNotEqual(default_node, new_node)

  87. 

  88.     def test_eq_negated(self):

  89.         node = Node(negated=False)

  90.         negated = Node(negated=True)

  91.         self.assertNotEqual(negated, node)
About Sign Up Leave Feedback Contact