Skip to content
Snippets Groups Projects

Update test_kg_maker.py

Closed Shufei Du requested to merge dushufei5-main-patch-94200 into main
1 file
+ 6
4
Compare changes
  • Side-by-side
  • Inline
+ 6
4
@@ -3,10 +3,12 @@ import pandas as pd
import sys
import os
import networkx as nx
import pandas.testing as pdt
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
from src.kg_maker import KgMaker
class TestCalculateMetrics(unittest.TestCase):
@classmethod
@@ -90,7 +92,7 @@ class TestGroupMetadata(unittest.TestCase):
})#have the same activity value
self.meta_data_same=KgMaker(self.df_same)
self.df=self.df_same=pd.DataFrame({
self.df=pd.DataFrame({
"case id":[1,2,3],
"activity":["activity1","activity2","activity3"],
"timestamp":["2024-05-01","2024-05-02","2024-05-03"],
@@ -99,15 +101,15 @@ class TestGroupMetadata(unittest.TestCase):
})#no same group
self.meta_data=KgMaker(self.df)
def test_add_meta_data(self):
result_df_same=self.meta_data_same.add_meta_data()
def test_group_meta_data(self):
result_df_same=self.meta_data_same.group_meta_data()
expected_df_same = pd.DataFrame({
"activity": ["activity1", "activity2"],
"meta1": ["meta1_value1", "meta1_value2"],
"meta2": ["meta2_value4", "meta2_value5"]
})
result_df=self.meta_data.add_meta_data()
result_df=self.meta_data.group_meta_data()
expected_df=pd.DataFrame({
"activity":["activity1","activity2","activity3"],
"meta1":["meta1_value1","meta1_value2","meta1_value3"],
Loading