code: def test_max_tree_sensitivity_square_sum_every_step_part(
      self, max_part
  ):
      steps, min_separation = 8, 0
      assert max_part > steps
      # If a sample will appear in every leaf node, we can infer the total
      # sensitivity by adding all the nodes.
      expected = 120
      self.assertEqual(
          expected,
          tree_aggregation_accountant._max_tree_sensitivity_square_sum(
              max_part, min_separation, steps
          ),
      )
  
file_path: privacy/tensorflow_privacy/privacy/analysis/tree_aggregation_accountant_test.py
imports: from absl.testing import parameterized
  import dp_accounting
  import tensorflow
  from tensorflow_privacy.privacy.analysis import tree_aggregation_accountant
label_desc: Use image node_id 8 for calling the TreeAggregationTest obj's underlying member method code with example usage: obj.test_max_tree_sensitivity_square_sum_every_step_part(max_part) without return types