import tensorflow as tf

from training_pipeline.models import preprocessing


class PreprocessingTest(tf.test.TestCase):
    def testPreprocessingFn(self):
        self.assertTrue(callable(preprocessing.preprocessing_fn))


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