def test_property_nx_graph(self):
q_graph = create_diamond_plus()
service = QuestionGraphService(q_graph)
nx_graph = service.nx_graph
self.assertIsInstance(nx_graph, MultiDiGraph)
self.assertEqual(len(nx_graph.nodes), 7)
def test_property_reachable_questions(self):
q_graph = create_diamond_plus()
service = QuestionGraphService(q_graph)
questions = service.reachable_questions
self.assertEqual(len(questions), 5)
self.assertEqual({q.analysis_key for q in questions}, set(f'q{n}' for n in range(1, 6)))
def test_build_nx_graph_no_choices_predefined(self):
q_graph = create_diamond_plus()
service = QuestionGraphService(q_graph)
edges = service._get_edges(q_graph)
nx_graph = service._build_nx_graph(q_graph, edges)
self.assertIsInstance(nx_graph, MultiDiGraph)
self.assertEqual(len(nx_graph.nodes), 7)
def test_get_endpoint_questions(self):
q_graph = create_diamond_plus()
service = QuestionGraphService(q_graph)
service.refresh_from_db()
endpoints_by_id = service.endpoint_questions
self.assertEqual(len(endpoints_by_id), 1)
question = list(endpoints_by_id.values())[0]
self.assertEqual(question.analysis_key, 'q5')
def test_get_reachable_questions(self):
q_graph = create_diamond_plus()
service = QuestionGraphService(q_graph)
edges = service._get_edges(q_graph)
nx_graph = service._build_nx_graph(q_graph, edges)
questions = service._get_reachable_questions(nx_graph, q_graph)
self.assertEqual(len(questions), 5)
self.assertEqual({q.analysis_key for q in questions.values()}, set(f'q{n}' for n in range(1, 6)))
def test_refresh_from_db(self):
q_graph = create_diamond_plus()
service = QuestionGraphService(q_graph)
service.refresh_from_db()
self.assertEqual(len(service._edges), 6)
self.assertIsInstance(service._nx_graph, MultiDiGraph)
self.assertEqual(len(service._nx_graph.nodes), 7)
self.assertEqual(len(service._questions), 7)
self.assertEqual(len(service._questions_by_id), 7)
def test_get_edges(self):
q_graph = create_diamond_plus()
service = QuestionGraphService(q_graph)
edges = service._get_edges(q_graph)
self.assertEqual(len(edges), 6)
def __init__(self, session):
self.session = session
self.question_graph_service = QuestionGraphService(
session.questionnaire.graph)
self.answer_service = AnswerService
self._path_validation_errors_by_uuid = {}