def test_gan_utility_funcs(simple_gan: GAN): xy_shp = simple_gan_z_shape[1:] x = np.zeros(xy_shp, dtype=np.float32) y = np.zeros(xy_shp, dtype=np.float32) simple_gan.interpolate(x, y) z_point = simple_gan.random_z_point() neighbors = simple_gan.neighborhood(z_point, std=0.05) diff = np.stack([neighbors[0]]*len(neighbors)) - neighbors assert np.abs(diff).mean() < 0.1