UnFlow/src/test.py

import tensorflow as tf
#from e2eflow.test.test_image_warp import ImageWarpTest
#from e2eflow.test.test_losses import LossesTest

#from e2eflow.test.ops.backward_warp import BackwardWarpTest
#from e2eflow.test.ops.forward_warp import ForwardWarpTest
#from e2eflow.test.ops.downsample import DownsampleTest
from e2eflow.test.ops.correlation import CorrelationTest


if __name__ == "__main__":
    tf.test.main()