@@ -47,8 +47,7 @@ def test_expression_helper(self):
4747
4848 # Verify expression is set in nested path
4949 assert (
50- proto .calculated_channel_configuration .query_configuration .sel .expression
51- == "$1 + $2"
50+ proto .calculated_channel_configuration .query_configuration .sel .expression == "$1 + $2"
5251 )
5352 assert "query_configuration" in mask .paths
5453
@@ -66,9 +65,7 @@ def test_expression_channel_references_helper(self):
6665 proto , mask = update .to_proto_with_mask ()
6766
6867 # Verify channel references are converted
69- refs = (
70- proto .calculated_channel_configuration .query_configuration .sel .expression_channel_references
71- )
68+ refs = proto .calculated_channel_configuration .query_configuration .sel .expression_channel_references
7269 assert len (refs ) == 2
7370 assert refs [0 ].channel_reference == "$1"
7471 assert refs [0 ].channel_identifier == "channel1"
@@ -110,10 +107,7 @@ def test_all_assets_helper(self):
110107 proto , mask = update .to_proto_with_mask ()
111108
112109 # Verify all_assets is set in nested path
113- assert (
114- proto .calculated_channel_configuration .asset_configuration .all_assets
115- is True
116- )
110+ assert proto .calculated_channel_configuration .asset_configuration .all_assets is True
117111 # Verify update_field is in mask (same as tag_ids and asset_ids)
118112 assert "asset_configuration" in mask .paths
119113
@@ -153,9 +147,7 @@ def test_expression_validator_rejects_references_without_expression(self):
153147 ):
154148 CalculatedChannelUpdate (
155149 expression_channel_references = [
156- ChannelReference (
157- channel_reference = "$1" , channel_identifier = "channel1"
158- ),
150+ ChannelReference (channel_reference = "$1" , channel_identifier = "channel1" ),
159151 ],
160152 )
161153
@@ -210,9 +202,7 @@ def mock_calculated_channel(mock_client):
210202class TestCalculatedChannel :
211203 """Unit tests for CalculatedChannel model - tests properties and methods."""
212204
213- def test_archive_calls_client_and_updates_self (
214- self , mock_calculated_channel , mock_client
215- ):
205+ def test_archive_calls_client_and_updates_self (self , mock_calculated_channel , mock_client ):
216206 """Test that archive() calls client.calculated_channels.archive and calls _update."""
217207 archived_calc_channel = MagicMock ()
218208 archived_calc_channel .is_archived = True
@@ -234,9 +224,7 @@ def test_archive_calls_client_and_updates_self(
234224 # Verify it returns self
235225 assert result is mock_calculated_channel
236226
237- def test_unarchive_calls_client_and_updates_self (
238- self , mock_calculated_channel , mock_client
239- ):
227+ def test_unarchive_calls_client_and_updates_self (self , mock_calculated_channel , mock_client ):
240228 """Test that unarchive() calls client.calculated_channels.unarchive and calls _update."""
241229 unarchived_calc_channel = MagicMock ()
242230 unarchived_calc_channel .is_archived = False
@@ -258,9 +246,7 @@ def test_unarchive_calls_client_and_updates_self(
258246 # Verify it returns self
259247 assert result is mock_calculated_channel
260248
261- def test_update_calls_client_and_updates_self (
262- self , mock_calculated_channel , mock_client
263- ):
249+ def test_update_calls_client_and_updates_self (self , mock_calculated_channel , mock_client ):
264250 """Test that update() calls client.calculated_channels.update and calls _update."""
265251 updated_calc_channel = MagicMock ()
266252 updated_calc_channel .description = "Updated description"
0 commit comments