code: def test_max_tree_sensitivity_square_sum_every_step_part2(
      self, steps
  ):
      max_participation, min_separation = 2, 0
      # If a sample will appear twice, the worst case is to put the two nodes at
      # consecutive nodes of the deepest subtree.
      steps_bin = bin(steps)[2:]
      depth = len(steps_bin) - 1
      expected = 2 + 4 * depth
      self.assertEqual(
          expected,
          tree_aggregation_accountant._max_tree_sensitivity_square_sum(
              max_participation, 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 9 for calling the TreeAggregationTest obj's underlying member method code with example usage: obj.test_max_tree_sensitivity_square_sum_every_step_part2(steps) without return types