Skip to content
Snippets Groups Projects

Update test_kg_maker.py

Merged Shufei Du requested to merge dushufei5-main-patch-29322 into main
1 file
+ 5
4
Compare changes
  • Side-by-side
  • Inline
+ 5
4
@@ -88,7 +88,8 @@ class TestGroupMetadata(unittest.TestCase):
"meta1":["meta1_value1","meta1_value2","meta1_value3"],
"meta2": ["meta2_value4", "meta2_value5", "meta2_value6"]
})#have the same activity value
self.meta_data_same=KgMaker(self.df_same)
self.meta_data_same=KgMaker(self.df_same,{})
self.df=self.df_same=pd.DataFrame({
"case id":[1,2,3],
@@ -97,17 +98,17 @@ class TestGroupMetadata(unittest.TestCase):
"meta1":["meta1_value1","meta1_value2","meta1_value3"],
"meta2": ["meta2_value4", "meta2_value5", "meta2_value6"]
})#no same group
self.meta_data=KgMaker(self.df)
self.meta_data=KgMaker(self.df,{})
def test_add_meta_data(self):
result_df_same=self.meta_data_same.add_meta_data()
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