Commit 2d6e9bfc authored by Jay's avatar Jay
Browse files

Updates to tests for code coverage

parent c71fcbc0

.coveragerc

+2 −12
Changes for .coveragerc: 2 added lines, 12 removed lines.
Original line number Diff line number Diff line
@@ -2,19 +2,9 @@ 
source = autocnet

[report]

omit =

    autocnet/fileio/ControlNetFileV0002_pb2.py

    autocnet/vis/graph_view.py

    autocnet/fileio/sqlalchemy_json/*

    autocnet/fileio/io_multibandimager.py

    autocnet/fileio/io_moon_minerology_mapper.py

    autocnet/fileio/header_parser.py

    autocnet/fileio/io_ccs.py

    autocnet/fileio/io_jsc.py

    autocnet/fileio/io_edr.py

    autocnet/fileio/lookup.py

    autocnet/fileio/utils.py

    autocnet/utils/folds.py

    autocnet/spectral/*

    autocnet/__init__.py

    autocnet/matcher/cuda_*

    bin/*

    */tests/*

exclude_lines =

autocnet/matcher/feature_extractor.py

+2 −2
Changes for autocnet/matcher/feature_extractor.py: 2 added lines, 2 removed lines.
Original line number Diff line number Diff line
@@ -9,7 +9,7 @@ from scipy.misc import bytescale 
try:

    import cyvlfeat as vl

    vlfeat = True

except Exception:

except Exception:  # pragma: no cover

    vlfeat = False

    pass
@@ -46,7 +46,7 @@ def extract_features(array, method='orb', extractor_parameters={}): 
                 'orb': cv2.ORB_create}



    if method == 'vlfeat' and vlfeat != True:

        print('VLFeat is not available.  Please install vlfeat or use a different extractor.')

        raise ImportError('VLFeat is not available.  Please install vlfeat or use a different extractor.')



    if  method == 'vlfeat':

        keypoint_objs, descriptors  = vl.sift.sift(array,

autocnet/matcher/tests/test_feature_extractor.py

+8 −2
Changes for autocnet/matcher/tests/test_feature_extractor.py: 8 added lines, 2 removed lines.
Original line number Diff line number Diff line
@@ -10,8 +10,6 @@ import sys 
from .. import feature_extractor

from plio.io import io_gdal



sys.path.insert(0, os.path.abspath('..'))





class TestFeatureExtractor(unittest.TestCase):
@@ -37,3 +35,11 @@ class TestFeatureExtractor(unittest.TestCase): 
                                                              extractor_parameters={})

        self.assertIsInstance(kps, pd.DataFrame)

        self.assertEqual(descriptors.dtype, np.float32)



    def test_extract_vlfeat_without(self):

        feature_extractor.vlfeat = False

        with self.assertRaises(ImportError):

            kps, descriptors = feature_extractor.extract_features(self.data_array,

                                                                  method='vlfeat',

                                                                  extractor_parameters={})

        feature_extractor.vlfeat = True