def test_get_class_data(self):
kb = KnowledgeBase(None, '..\\..\\test_data\\Test_Data_AB.csv')
kb.add_data('..\\..\\test_data\\Test_Data_C.csv')
print(kb.get_data())
print()
print(kb.get_class_data(['A', 'C']))
print()
print(kb.get_class_data(['B']))
print(kb.get_class_data(['Z']))
print(kb.get_class_data(['A', 'B', 'Z']))
def test_add_data(self):
kb = KnowledgeBase(None, '..\\..\\test_data\\Test_Data_AB.csv')
print(kb.get_data())
print()
kb.add_data('..\\..\\test_data\\Test_Data_C.csv')
print(kb.get_data())
def test_update_data(self):
kb = KnowledgeBase(None, '..\\..\\test_data\\Test_Data_AB.csv')
print(kb.get_data())
print()
kb.update_data('..\\..\\test_data\\test_data_A_update.csv')
print(kb.get_data())
from bayesian_network import BayesianNetwork
from knowledge_base import KnowledgeBase
from graph_builder import GraphBuilder
if __name__ == "__main__":
_data_file_path = '..\\ExcelFiles\\courses_and_grades.csv'
_relations_file_path = '..\\..\\Data\\combined_course_structure.csv'
knowledge_base = KnowledgeBase(_relations_file_path, _data_file_path)
builder = GraphBuilder()
builder = builder.build_nodes(list(knowledge_base.get_data().columns))
builder = builder.add_parents(knowledge_base.get_relations())
builder = builder.add_children()
builder = builder.build_edges()
graph = builder.build_graph()
nodes = graph.get_nodes()
# bayes_net = BayesianNetwork(knowledge_base, graph)
# print(bayes_net.get_graph().get_node('Calculus and Analytic Geometry I').get_parents())
