code: def test_in_first(self) -> None:
      # TODO: Make this work with TTP provider
      cfg.mpc.provider = "TFP"
  
      # This will cause the parent process to init with world-size 1
      crypten.init()
      self.assertEqual(comm.get().get_world_size(), 1)
  
      # This will fork 2 children which will have to init with world-size 2
      self.assertEqual(test_rank_func(), [0, 1])
  
      # Make sure everything is the same in the parent
      self.assertEqual(comm.get().get_world_size(), 1)
  
imports: import unittest
  import crypten
  import crypten.communicator
  import crypten.mpc
  import torch
  from crypten.config import cfg
label_desc: Use image node_id 4 for calling the TestContext obj's underlying member method code with example usage: obj.test_in_first() without return types