test_read_reverse_dict.py 6.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165
  1. import unittest
  2. from modelverse_state.main import ModelverseState
  3. from modelverse_state import status
  4. class TestReadReverseDict(unittest.TestCase):
  5. def setUp(self):
  6. self.mvs = ModelverseState()
  7. def test_read_reverse_dict_no_exists(self):
  8. l, r = self.mvs.read_reverse_dict(-1, "abc")
  9. self.assertEquals(l, None)
  10. self.assertEquals(r, status.FAIL_RRDICT_UNKNOWN)
  11. def test_read_reverse_dict_not_found_node(self):
  12. a, r = self.mvs.create_node()
  13. self.assertEquals(r, status.SUCCESS)
  14. # Passing data is not enforced, as the data will be interpreted if necessary
  15. l, r = self.mvs.read_reverse_dict(a, "abc")
  16. self.assertEquals(l, None)
  17. self.assertEquals(r, status.FAIL_RRDICT_NOT_FOUND)
  18. def test_read_reverse_dict_not_found_nodevalue(self):
  19. a, r = self.mvs.create_nodevalue(1)
  20. self.assertEquals(r, status.SUCCESS)
  21. # Passing data is not enforced, as the data will be interpreted if necessary
  22. l, r = self.mvs.read_reverse_dict(a, "abc")
  23. self.assertEquals(l, None)
  24. self.assertEquals(r, status.FAIL_RRDICT_NOT_FOUND)
  25. def test_read_reverse_dict_not_found_edge(self):
  26. a, r = self.mvs.create_node()
  27. self.assertEquals(r, status.SUCCESS)
  28. b, r = self.mvs.create_node()
  29. self.assertEquals(r, status.SUCCESS)
  30. c, r = self.mvs.create_edge(a, b)
  31. self.assertEquals(r, status.SUCCESS)
  32. # Passing data is not enforced, as the data will be interpreted if necessary
  33. l, r = self.mvs.read_reverse_dict(c, "abc")
  34. self.assertEquals(l, None)
  35. self.assertEquals(r, status.FAIL_RRDICT_NOT_FOUND)
  36. def test_read_reverse_dict_no_primitive(self):
  37. a, r = self.mvs.create_node()
  38. self.assertEquals(r, status.SUCCESS)
  39. # Passing data is not enforced, as the data will be interpreted if necessary
  40. l, r = self.mvs.read_reverse_dict(a, a)
  41. self.assertEquals(l, None)
  42. self.assertEquals(r, status.FAIL_RRDICT_NOT_FOUND)
  43. def test_read_reverse_dict_node_simple(self):
  44. a, r = self.mvs.create_node()
  45. self.assertEquals(r, status.SUCCESS)
  46. b, r = self.mvs.create_node()
  47. self.assertEquals(r, status.SUCCESS)
  48. c, r = self.mvs.create_nodevalue("f")
  49. self.assertEquals(r, status.SUCCESS)
  50. d, r = self.mvs.create_edge(a, b)
  51. self.assertEquals(r, status.SUCCESS)
  52. e, r = self.mvs.create_edge(d, c)
  53. self.assertEquals(r, status.SUCCESS)
  54. l, r = self.mvs.read_reverse_dict(b, "f")
  55. self.assertEquals(r, status.SUCCESS)
  56. self.assertEquals(l, [a])
  57. def test_read_reverse_dict_no_match(self):
  58. a, r = self.mvs.create_node()
  59. self.assertEquals(r, status.SUCCESS)
  60. b, r = self.mvs.create_node()
  61. self.assertEquals(r, status.SUCCESS)
  62. c, r = self.mvs.create_nodevalue("g")
  63. self.assertEquals(r, status.SUCCESS)
  64. d, r = self.mvs.create_edge(a, b)
  65. self.assertEquals(r, status.SUCCESS)
  66. e, r = self.mvs.create_edge(d, c)
  67. self.assertEquals(r, status.SUCCESS)
  68. l, r = self.mvs.read_reverse_dict(b, "f")
  69. self.assertEquals(r, status.FAIL_RRDICT_NOT_FOUND)
  70. self.assertEquals(l, None)
  71. def test_read_reverse_dict_node_multi(self):
  72. a, r = self.mvs.create_node()
  73. self.assertEquals(r, status.SUCCESS)
  74. b, r = self.mvs.create_node()
  75. self.assertEquals(r, status.SUCCESS)
  76. c, r = self.mvs.create_nodevalue("f")
  77. self.assertEquals(r, status.SUCCESS)
  78. d, r = self.mvs.create_edge(a, b)
  79. self.assertEquals(r, status.SUCCESS)
  80. e, r = self.mvs.create_edge(d, c)
  81. self.assertEquals(r, status.SUCCESS)
  82. g, r = self.mvs.create_node()
  83. self.assertEquals(r, status.SUCCESS)
  84. h, r = self.mvs.create_nodevalue("k")
  85. self.assertEquals(r, status.SUCCESS)
  86. i, r = self.mvs.create_edge(a, g)
  87. self.assertEquals(r, status.SUCCESS)
  88. j, r = self.mvs.create_edge(i, h)
  89. self.assertEquals(r, status.SUCCESS)
  90. l, r = self.mvs.read_reverse_dict(b, "f")
  91. self.assertEquals(r, status.SUCCESS)
  92. self.assertEquals(l, [a])
  93. l, r = self.mvs.read_reverse_dict(g, "k")
  94. self.assertEquals(r, status.SUCCESS)
  95. self.assertEquals(l, [a])
  96. l, r = self.mvs.read_reverse_dict(a, "l")
  97. self.assertEquals(r, status.FAIL_RRDICT_NOT_FOUND)
  98. self.assertEquals(l, None)
  99. def test_read_reverse_dict_node_multi_ambiguous(self):
  100. a, r = self.mvs.create_node()
  101. self.assertEquals(r, status.SUCCESS)
  102. b, r = self.mvs.create_node()
  103. self.assertEquals(r, status.SUCCESS)
  104. c, r = self.mvs.create_nodevalue("f")
  105. self.assertEquals(r, status.SUCCESS)
  106. d, r = self.mvs.create_edge(b, a)
  107. self.assertEquals(r, status.SUCCESS)
  108. e, r = self.mvs.create_edge(d, c)
  109. self.assertEquals(r, status.SUCCESS)
  110. g, r = self.mvs.create_node()
  111. self.assertEquals(r, status.SUCCESS)
  112. h, r = self.mvs.create_nodevalue("f")
  113. self.assertEquals(r, status.SUCCESS)
  114. i, r = self.mvs.create_edge(g, a)
  115. self.assertEquals(r, status.SUCCESS)
  116. j, r = self.mvs.create_edge(i, h)
  117. self.assertEquals(r, status.SUCCESS)
  118. l, r = self.mvs.read_reverse_dict(a, "f")
  119. self.assertEquals(r, status.SUCCESS)
  120. self.assertEquals(sorted(l), sorted([b, g]))
  121. def test_read_reverse_dict_node_uncertain(self):
  122. a, r = self.mvs.create_node()
  123. self.assertEquals(r, status.SUCCESS)
  124. b, r = self.mvs.create_node()
  125. self.assertEquals(r, status.SUCCESS)
  126. c, r = self.mvs.create_nodevalue("f")
  127. self.assertEquals(r, status.SUCCESS)
  128. d, r = self.mvs.create_edge(a, b)
  129. self.assertEquals(r, status.SUCCESS)
  130. e, r = self.mvs.create_edge(d, c)
  131. self.assertEquals(r, status.SUCCESS)
  132. h, r = self.mvs.create_nodevalue("g")
  133. self.assertEquals(r, status.SUCCESS)
  134. i, r = self.mvs.create_edge(d, h)
  135. self.assertEquals(r, status.SUCCESS)
  136. l, r = self.mvs.read_reverse_dict(b, "f")
  137. self.assertEquals(r, status.FAIL_RRDICT_UNCERTAIN)
  138. self.assertEquals(l, None)