def test_dataset_paths(self):
w.FLAGS.batch_size = 2
with self.test_session() as sess:
np.set_printoptions(threshold=np.nan)
x, _ = w.dataset_paths(['assets/cat.png'])
sess.run(tf.tables_initializer())
self.assertTupleEqual(x.eval().shape, (1, 300, 300, 4))
def ds():
return w.dataset_paths([
'assets/cat-selection.png',
])
def dm():
return w.dataset_paths([
'assets/empty.png',
])
def d_cherry():
return w.dataset_paths([
'assets/cat.png',
])