Asserting on conditions with tf.Assert()