import unittest
import sys
sys.path.insert(0,"C:/Users/SUMANTH C/Desktop/DR/src")
from features_input import *

class ImportError(unittest.TestCase):

    def test_import(self):
        sys.path.insert(0, "C:/Users/SUMANTH C/Desktop/DR/src")
        from features_input import VGG16Conv


    def test_features(self):
        img_path = "E:\\DR\\datasets\\filtered_dataset\\train001\\"
        initial = 0
        final = 1

        #self.assertGreater(final, initial)
        features_input = VGG16Conv(img_path, initial, final)
        print(features_input.shape)


if __name__=='__main__':
    unittest.main()