def METHOD_NAME(self):
    for tf_tokenizer in self.tf_tokenizers:
        model = ModelToSave(tokenizer=tf_tokenizer)
        test_inputs = tf.convert_to_tensor(self.test_sentences)
        out = model(test_inputs)  # Build model with some sample inputs
        with TemporaryDirectory() as tempdir:
            save_path = Path(tempdir) / "saved.model"
            model.save(save_path)
            loaded_model = tf.keras.models.load_model(save_path)
        loaded_output = loaded_model(test_inputs)
        # We may see small differences because the loaded model is compiled, so we need an epsilon for the test
        self.assertLessEqual(tf.reduce_max(tf.abs(out - loaded_output)), 1e-5)