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())
