def METHOD_NAME(self, model, expected_inputs, expected_outputs):
    interpreter = tf.lite.Interpreter(model_content=model)
    inputs = [
        input_["shape"].tolist() for input_ in interpreter.get_input_details()
    ]
    outputs = [
        output["shape"].tolist() for output in interpreter.get_output_details()
    ]
    self.assertEqual(inputs, [list(dims) for dims in expected_inputs])
    self.assertEqual(outputs, [list(dims) for dims in expected_outputs])