@@ -64,8 +64,7 @@ def setUp(self) -> None:
64
64
self .users_in_any_filter_group = list (set (self .users_in_group1 + self .users_in_group2 ))
65
65
self .all_users = list (self .users .values ())
66
66
self .users_not_in_filter_groups = [
67
- user for user in self .users .values ()
68
- if user not in self .users_in_any_filter_group
67
+ user for user in self .users .values () if user not in self .users_in_any_filter_group
69
68
]
70
69
71
70
# Assign users to groups
@@ -87,12 +86,12 @@ def test_no_filter_groups(self):
87
86
self .assertGreaterEqual (groups .count (), len (self .all_groups ))
88
87
89
88
# Verify all users are included
90
- user_pks = list (users .values_list ('pk' , flat = True ))
89
+ user_pks = list (users .values_list ("pk" , flat = True ))
91
90
for user in self .all_users :
92
91
self .assertIn (user .pk , user_pks )
93
92
94
93
# Verify all test groups are included
95
- group_pks = list (groups .values_list ('pk' , flat = True ))
94
+ group_pks = list (groups .values_list ("pk" , flat = True ))
96
95
for group in self .all_groups :
97
96
self .assertIn (group .pk , group_pks )
98
97
@@ -108,17 +107,17 @@ def test_single_filter_group(self):
108
107
self .assertEqual (groups .count (), 1 )
109
108
110
109
# Verify only users in filter_group1 are included
111
- user_pks = set (users .values_list ('pk' , flat = True ))
110
+ user_pks = set (users .values_list ("pk" , flat = True ))
112
111
for user in self .all_users :
113
- if ( user in self .users_in_group1 ) :
112
+ if user in self .users_in_group1 :
114
113
self .assertIn (user .pk , user_pks )
115
114
else :
116
115
self .assertNotIn (user .pk , user_pks )
117
116
118
117
# Verify only filter_group1 is included
119
- group_pks = list (groups .values_list ('pk' , flat = True ))
118
+ group_pks = list (groups .values_list ("pk" , flat = True ))
120
119
for group in self .all_groups :
121
- if ( group == self .filter_group1 ) :
120
+ if group == self .filter_group1 :
122
121
self .assertIn (group .pk , group_pks )
123
122
else :
124
123
self .assertNotIn (group .pk , group_pks )
@@ -135,17 +134,17 @@ def test_multiple_filter_groups(self):
135
134
self .assertEqual (groups .count (), len (self .filter_groups ))
136
135
137
136
# Verify users in either filter_group1 or filter_group2 are included
138
- user_pks = set (users .values_list ('pk' , flat = True ))
137
+ user_pks = set (users .values_list ("pk" , flat = True ))
139
138
for user in self .all_users :
140
- if ( user in self .users_in_any_filter_group ) :
139
+ if user in self .users_in_any_filter_group :
141
140
self .assertIn (user .pk , user_pks )
142
141
else :
143
142
self .assertNotIn (user .pk , user_pks )
144
143
145
144
# Verify only the filter groups are included
146
- group_pks = list (groups .values_list ('pk' , flat = True ))
145
+ group_pks = list (groups .values_list ("pk" , flat = True ))
147
146
for group in self .all_groups :
148
- if ( group in self .filter_groups ) :
147
+ if group in self .filter_groups :
149
148
self .assertIn (group .pk , group_pks )
150
149
else :
151
150
self .assertNotIn (group .pk , group_pks )
@@ -214,6 +213,14 @@ def test_sync_with_filter_groups(self, mock: Mocker):
214
213
group_posts += 1
215
214
216
215
# Verify number of synchronized users
217
- self .assertEqual (user_posts , len (self .users_in_any_filter_group ), f"Expected { len (self .users_in_any_filter_group )} users to be synchronized" )
216
+ self .assertEqual (
217
+ user_posts ,
218
+ len (self .users_in_any_filter_group ),
219
+ f"Expected { len (self .users_in_any_filter_group )} users to be synchronized" ,
220
+ )
218
221
# Verify number of synchronized groups
219
- self .assertEqual (group_posts , len (self .filter_groups ), f"Expected { len (self .users_in_any_filter_group )} groups to be synchronized" )
222
+ self .assertEqual (
223
+ group_posts ,
224
+ len (self .filter_groups ),
225
+ f"Expected { len (self .users_in_any_filter_group )} groups to be synchronized" ,
226
+ )
0 commit comments