gan_test.py

def test_gen_training():
  model = Model
  sess = tf.Session()
  gen_vars = tf.get_collection(tf.GraphKeys.VARIABLES, scope='gen')
  des_vars = tf.get_collection(tf.GraphKeys.VARIABLES, scope='des')
  before_gen = sess.run(gen_vars)
  before_des = sess.run(des_vars)
  # Train the generator.
  sess.run(model.train_gen)
  after_gen = sess.run(gen_vars)
  after_des = sess.run(des_vars)
  # Make sure the generator variables changed.
  for b,a in zip(before_gen, after_gen):
    assert (a != b).any()
  # Make sure descriminator did NOT change.
  for b,a in zip(before_des, after_des):
    assert (a == b).all()