updated test

1 files changed, 2 insertions, 0 deletions

diff --git a/test/biggan_test.py b/test/biggan_test.py
index db3e59f..b1a5b85 100644
--- a/test/biggan_test.py
+++ b/test/biggan_test.py
@@ -2,6 +2,7 @@ import unittest
 from scipy.stats import truncnorm
 import numpy as np
 from gantools import biggan
+from gantools import image_utils
 
 def create_random_input(dim_z, vocab_size, batch_size=1, truncation = 0.5, rand_seed = 123):
     def one_hot(index, dim):
@@ -20,6 +21,7 @@ class TestBigGAN(unittest.TestCase):
         gan = biggan.BigGAN()
         vectors, labels, truncation = create_random_input(gan.dim_z, gan.vocab_size)
         ims = gan.sample(vectors, labels, truncation)
+        image_utils.save_images(ims)
 
 
 if __name__ == '__main__':