def METHOD_NAME(self):
    context = create_model_context(data_size=16, batch_size=2)
    self.assertFalse(context.find_unused_parameters)
    context = create_model_context(data_size=16, batch_size=2, extra_args={"find_unused_parameters": True})
    self.assertTrue(context.find_unused_parameters)