code: def test_forget_batch():
      # Check stored bounding boxes and leaf counts after forgetting points
      for index in indexes:
          forgotten = tree.forget_point(index)
          branches = []
          tree.map_branches(
              tree.root, op=tree._get_nodes, stack=branches
          )
          for branch in branches:
              leafcount = tree._count_leaves(branch)
              try:
                  assert leafcount == branch.n
              except:
                  print(forgotten.x)
                  print("Computed:\n", leafcount)
                  print("Stored:\n", branch.n)
                  raise
              bbox = tree.get_bbox(branch)
              try:
                  assert np.allclose(bbox, branch.b)
              except:
                  print(forgotten.x)
                  print("Computed:\n", bbox)
                  print("Stored:\n", branch.b)
                  raise
  
file_path: rrcf/test/test_rrcf.py
imports: import json
  import numpy
  import rrcf
label_desc: Use image node_id 5 for calling a global function with example usage: test_forget_batch() without return types