def METHOD_NAME(self):
    for add_data_parallel in [False, True]:
        model, state_dict = self.create_complex_model()
        if add_data_parallel:
            model = nn.DataParallel(model)
        model_sd = model.state_dict()
        sd_to_load = align_and_update_state_dicts(model_sd, state_dict)
        model.load_state_dict(sd_to_load)
        for loaded, stored in zip(model_sd.values(), state_dict.values()):
            # different tensor references
            self.assertFalse(id(loaded) == id(stored))
            # same content
            self.assertTrue(loaded.to(stored).equal(stored))