def test_fsdp_utils_size_based_policy():
    """
    Test constructor for underlying FSDP policy is called with correct arguments
    """
    min_num_params = 100
    policy = SizeBasedWrapPolicy(min_num_params=min_num_params)
    policy = policy.get_policy()
    assert policy.keywords["min_num_params"] == min_num_params

    Test constructor for underlying FSDP policy is called with correct arguments
    
coremltools/test/optimize/torch/test_utils/test_fsdp_utils.py