Skip to content

Commit 61a0ec8

Browse files
committed
added dip tests
1 parent 576e308 commit 61a0ec8

File tree

1 file changed

+43
-0
lines changed

1 file changed

+43
-0
lines changed
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,43 @@
1+
from pytti.image_models.differentiable_image import DifferentiableImage
2+
from pytti.image_models.ema import EMAImage
3+
from pytti.image_models.pixel import PixelImage
4+
from pytti.image_models.rgb_image import RGBImage
5+
from pytti.image_models.vqgan import VQGANImage
6+
from pytti.image_models.deep_image_prior import DeepImagePrior
7+
8+
### DIP ###
9+
10+
11+
def test_dip_init():
12+
obj = DeepImagePrior(512, 512)
13+
assert obj
14+
15+
16+
def test_dip_update():
17+
obj = DeepImagePrior(512, 512)
18+
obj.update()
19+
20+
21+
def test_dip_forward():
22+
obj = DeepImagePrior(512, 512)
23+
obj.forward()
24+
25+
26+
def test_dip_decode_training_tensor():
27+
obj = DeepImagePrior(512, 512)
28+
obj.decode_training_tensor()
29+
30+
31+
def test_dip_decode_tensor():
32+
obj = DeepImagePrior(512, 512)
33+
obj.decode_tensor()
34+
35+
36+
def test_dip_clone():
37+
obj = DeepImagePrior(512, 512)
38+
obj.clone()
39+
40+
41+
def test_dip_get_latent_tensor():
42+
obj = DeepImagePrior(10, 10)
43+
obj.get_latent_tensor()

0 commit comments

Comments
 (0)