mltest_example.py

import mltest

def test_my_model():
  # Make placeholders for input into the model
  input_image = tf.placeholder(...)
  label = tf.placeholder(...)
  # Build your model
  model = Model(input_image, label)
  # Run the test suite!
  mltest.test_suite(
    model.logits,
    model.train_op,
    feed_dict={
      input_image: np.random.randn(...)
      label: np.random.randint(...)
    })