generated from TogetherCrew/python-library
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #20 from TogetherCrew/feat/include-actions
Feat/include actions
- Loading branch information
Showing
9 changed files
with
1,065 additions
and
17 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
|
|
@@ -6,7 +6,7 @@ | |
|
|
||
| setup( | ||
| name="tc-core-analyzer-lib", | ||
| version="1.1.0", | ||
| version="1.2.0", | ||
| author="Mohammad Amin Dadgar, TogetherCrew", | ||
| maintainer="Mohammad Amin Dadgar", | ||
| maintainer_email="[email protected]", | ||
|
|
||
100 changes: 100 additions & 0 deletions
100
tc_core_analyzer_lib/tests/integration/test_assess_mention.py
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,100 @@ | ||
| from unittest import TestCase | ||
|
|
||
| import numpy as np | ||
| from tc_core_analyzer_lib.assess_engagement import EngagementAssessment | ||
| from tc_core_analyzer_lib.utils.activity import DiscordActivity | ||
|
|
||
|
|
||
| class TestAssessMentions(TestCase): | ||
| def setUp(self) -> None: | ||
| self.window_days = 7 | ||
| self.action_params = { | ||
| "INT_THR": 1, | ||
| "UW_DEG_THR": 1, | ||
| "PAUSED_T_THR": 1, | ||
| "CON_T_THR": 4, | ||
| "CON_O_THR": 3, | ||
| "EDGE_STR_THR": 5, | ||
| "UW_THR_DEG_THR": 5, | ||
| "VITAL_T_THR": 4, | ||
| "VITAL_O_THR": 3, | ||
| "STILL_T_THR": 2, | ||
| "STILL_O_THR": 2, | ||
| "DROP_H_THR": 2, | ||
| "DROP_I_THR": 1, | ||
| } | ||
|
|
||
| def test_assess_single_reply(self): | ||
| acc_names = [] | ||
| acc_count = 5 | ||
| for i in range(5): | ||
| acc_names.append(f"user{i}") | ||
|
|
||
| acc_names = np.array(acc_names) | ||
|
|
||
| int_mat = { | ||
| DiscordActivity.Reply: np.zeros((acc_count, acc_count)), | ||
| DiscordActivity.Mention: np.zeros((acc_count, acc_count)), | ||
| DiscordActivity.Reaction: np.zeros((acc_count, acc_count)), | ||
| DiscordActivity.Lone_msg: np.zeros((acc_count, acc_count)), | ||
| DiscordActivity.Thread_msg: np.zeros((acc_count, acc_count)), | ||
| } | ||
|
|
||
| # user0 mentioning user1 | ||
| int_mat[DiscordActivity.Mention][0, 1] = 1 | ||
|
|
||
| activity_dict = { | ||
| "all_joined": {"0": set()}, | ||
| "all_joined_day": {"0": set()}, | ||
| "all_consistent": {}, | ||
| "all_vital": {}, | ||
| "all_active": {}, | ||
| "all_connected": {}, | ||
| "all_paused": {}, | ||
| "all_new_disengaged": {}, | ||
| "all_disengaged": {}, | ||
| "all_unpaused": {}, | ||
| "all_returned": {}, | ||
| "all_new_active": {}, | ||
| "all_still_active": {}, | ||
| "all_dropped": {}, | ||
| "all_disengaged_were_newly_active": {}, | ||
| "all_disengaged_were_consistently_active": {}, | ||
| "all_disengaged_were_vital": {}, | ||
| "all_lurker": {}, | ||
| "all_about_to_disengage": {}, | ||
| "all_disengaged_in_past": {}, | ||
| } | ||
| # window index | ||
| w_i = 0 | ||
|
|
||
| activities = [ | ||
| DiscordActivity.Reaction, | ||
| DiscordActivity.Mention, | ||
| DiscordActivity.Reply, | ||
| DiscordActivity.Lone_msg, | ||
| DiscordActivity.Thread_msg, | ||
| ] | ||
|
|
||
| engagement = EngagementAssessment( | ||
| activities=activities, | ||
| activities_ignore_0_axis=[ | ||
| DiscordActivity.Mention, | ||
| DiscordActivity.Reaction, | ||
| DiscordActivity.Reply, | ||
| ], | ||
| activities_ignore_1_axis=[], | ||
| ) | ||
|
|
||
| (_, *computed_activities) = engagement.compute( | ||
| int_mat=int_mat, | ||
| w_i=w_i, | ||
| acc_names=acc_names, | ||
| act_param=self.action_params, | ||
| WINDOW_D=self.window_days, | ||
| **activity_dict, | ||
| ) | ||
|
|
||
| computed_activities = dict(zip(activity_dict.keys(), computed_activities)) | ||
|
|
||
| self.assertEqual(computed_activities["all_active"], {"0": set(["user0"])}) |
100 changes: 100 additions & 0 deletions
100
tc_core_analyzer_lib/tests/integration/test_assess_reactions.py
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,100 @@ | ||
| from unittest import TestCase | ||
|
|
||
| import numpy as np | ||
| from tc_core_analyzer_lib.assess_engagement import EngagementAssessment | ||
| from tc_core_analyzer_lib.utils.activity import DiscordActivity | ||
|
|
||
|
|
||
| class TestAssessReactions(TestCase): | ||
| def setUp(self) -> None: | ||
| self.window_days = 7 | ||
| self.action_params = { | ||
| "INT_THR": 1, | ||
| "UW_DEG_THR": 1, | ||
| "PAUSED_T_THR": 1, | ||
| "CON_T_THR": 4, | ||
| "CON_O_THR": 3, | ||
| "EDGE_STR_THR": 5, | ||
| "UW_THR_DEG_THR": 5, | ||
| "VITAL_T_THR": 4, | ||
| "VITAL_O_THR": 3, | ||
| "STILL_T_THR": 2, | ||
| "STILL_O_THR": 2, | ||
| "DROP_H_THR": 2, | ||
| "DROP_I_THR": 1, | ||
| } | ||
|
|
||
| def test_assess_single_reaction(self): | ||
| acc_names = [] | ||
| acc_count = 5 | ||
| for i in range(5): | ||
| acc_names.append(f"user{i}") | ||
|
|
||
| acc_names = np.array(acc_names) | ||
|
|
||
| int_mat = { | ||
| DiscordActivity.Reply: np.zeros((acc_count, acc_count)), | ||
| DiscordActivity.Mention: np.zeros((acc_count, acc_count)), | ||
| DiscordActivity.Reaction: np.zeros((acc_count, acc_count)), | ||
| DiscordActivity.Lone_msg: np.zeros((acc_count, acc_count)), | ||
| DiscordActivity.Thread_msg: np.zeros((acc_count, acc_count)), | ||
| } | ||
|
|
||
| # user0 reacting to user1 | ||
| int_mat[DiscordActivity.Reaction][0, 1] = 1 | ||
|
|
||
| activity_dict = { | ||
| "all_joined": {"0": set()}, | ||
| "all_joined_day": {"0": set()}, | ||
| "all_consistent": {}, | ||
| "all_vital": {}, | ||
| "all_active": {}, | ||
| "all_connected": {}, | ||
| "all_paused": {}, | ||
| "all_new_disengaged": {}, | ||
| "all_disengaged": {}, | ||
| "all_unpaused": {}, | ||
| "all_returned": {}, | ||
| "all_new_active": {}, | ||
| "all_still_active": {}, | ||
| "all_dropped": {}, | ||
| "all_disengaged_were_newly_active": {}, | ||
| "all_disengaged_were_consistently_active": {}, | ||
| "all_disengaged_were_vital": {}, | ||
| "all_lurker": {}, | ||
| "all_about_to_disengage": {}, | ||
| "all_disengaged_in_past": {}, | ||
| } | ||
| # window index | ||
| w_i = 0 | ||
|
|
||
| activities = [ | ||
| DiscordActivity.Reaction, | ||
| DiscordActivity.Mention, | ||
| DiscordActivity.Reply, | ||
| DiscordActivity.Lone_msg, | ||
| DiscordActivity.Thread_msg, | ||
| ] | ||
|
|
||
| engagement = EngagementAssessment( | ||
| activities=activities, | ||
| activities_ignore_0_axis=[ | ||
| DiscordActivity.Mention, | ||
| DiscordActivity.Reaction, | ||
| DiscordActivity.Reply, | ||
| ], | ||
| activities_ignore_1_axis=[], | ||
| ) | ||
|
|
||
| (_, *computed_activities) = engagement.compute( | ||
| int_mat=int_mat, | ||
| w_i=w_i, | ||
| acc_names=acc_names, | ||
| act_param=self.action_params, | ||
| WINDOW_D=self.window_days, | ||
| **activity_dict, | ||
| ) | ||
|
|
||
| computed_activities = dict(zip(activity_dict.keys(), computed_activities)) | ||
|
|
||
| self.assertEqual(computed_activities["all_active"], {"0": set(["user0"])}) |
100 changes: 100 additions & 0 deletions
100
tc_core_analyzer_lib/tests/integration/test_assess_replies.py
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,100 @@ | ||
| from unittest import TestCase | ||
|
|
||
| import numpy as np | ||
| from tc_core_analyzer_lib.assess_engagement import EngagementAssessment | ||
| from tc_core_analyzer_lib.utils.activity import DiscordActivity | ||
|
|
||
|
|
||
| class TestAssessReplies(TestCase): | ||
| def setUp(self) -> None: | ||
| self.window_days = 7 | ||
| self.action_params = { | ||
| "INT_THR": 1, | ||
| "UW_DEG_THR": 1, | ||
| "PAUSED_T_THR": 1, | ||
| "CON_T_THR": 4, | ||
| "CON_O_THR": 3, | ||
| "EDGE_STR_THR": 5, | ||
| "UW_THR_DEG_THR": 5, | ||
| "VITAL_T_THR": 4, | ||
| "VITAL_O_THR": 3, | ||
| "STILL_T_THR": 2, | ||
| "STILL_O_THR": 2, | ||
| "DROP_H_THR": 2, | ||
| "DROP_I_THR": 1, | ||
| } | ||
|
|
||
| def test_assess_single_reply(self): | ||
| acc_names = [] | ||
| acc_count = 5 | ||
| for i in range(5): | ||
| acc_names.append(f"user{i}") | ||
|
|
||
| acc_names = np.array(acc_names) | ||
|
|
||
| int_mat = { | ||
| DiscordActivity.Reply: np.zeros((acc_count, acc_count)), | ||
| DiscordActivity.Mention: np.zeros((acc_count, acc_count)), | ||
| DiscordActivity.Reaction: np.zeros((acc_count, acc_count)), | ||
| DiscordActivity.Lone_msg: np.zeros((acc_count, acc_count)), | ||
| DiscordActivity.Thread_msg: np.zeros((acc_count, acc_count)), | ||
| } | ||
|
|
||
| # user0 replying user1 | ||
| int_mat[DiscordActivity.Reply][0, 1] = 1 | ||
|
|
||
| activity_dict = { | ||
| "all_joined": {"0": set()}, | ||
| "all_joined_day": {"0": set()}, | ||
| "all_consistent": {}, | ||
| "all_vital": {}, | ||
| "all_active": {}, | ||
| "all_connected": {}, | ||
| "all_paused": {}, | ||
| "all_new_disengaged": {}, | ||
| "all_disengaged": {}, | ||
| "all_unpaused": {}, | ||
| "all_returned": {}, | ||
| "all_new_active": {}, | ||
| "all_still_active": {}, | ||
| "all_dropped": {}, | ||
| "all_disengaged_were_newly_active": {}, | ||
| "all_disengaged_were_consistently_active": {}, | ||
| "all_disengaged_were_vital": {}, | ||
| "all_lurker": {}, | ||
| "all_about_to_disengage": {}, | ||
| "all_disengaged_in_past": {}, | ||
| } | ||
| # window index | ||
| w_i = 0 | ||
|
|
||
| activities = [ | ||
| DiscordActivity.Reaction, | ||
| DiscordActivity.Mention, | ||
| DiscordActivity.Reply, | ||
| DiscordActivity.Lone_msg, | ||
| DiscordActivity.Thread_msg, | ||
| ] | ||
|
|
||
| engagement = EngagementAssessment( | ||
| activities=activities, | ||
| activities_ignore_0_axis=[ | ||
| DiscordActivity.Mention, | ||
| DiscordActivity.Reaction, | ||
| DiscordActivity.Reply, | ||
| ], | ||
| activities_ignore_1_axis=[], | ||
| ) | ||
|
|
||
| (_, *computed_activities) = engagement.compute( | ||
| int_mat=int_mat, | ||
| w_i=w_i, | ||
| acc_names=acc_names, | ||
| act_param=self.action_params, | ||
| WINDOW_D=self.window_days, | ||
| **activity_dict, | ||
| ) | ||
|
|
||
| computed_activities = dict(zip(activity_dict.keys(), computed_activities)) | ||
|
|
||
| self.assertEqual(computed_activities["all_active"], {"0": set(["user0"])}) |
Oops, something went wrong.