def test_configuration(self, _mock_scd):
     i = ColorDescriptor_Image_csift(
         model_directory='test model dir',
         work_directory='test work dir',
         model_gen_descriptor_limit=123764,
         kmeans_k=42,
         flann_distance_metric='hik',
         flann_target_precision=0.92,
         flann_sample_fraction=0.71,
         flann_autotune=True,
         random_seed=7,
         use_spatial_pyramid=True,
         parallel=3,
     )
     for inst in configuration_test_helper(i):
         assert inst._model_dir == 'test model dir'
         assert inst._work_dir == 'test work dir'
         assert inst._model_gen_descriptor_limit == 123764
         assert inst._kmeans_k == 42
         assert inst._flann_distance_metric == 'hik'
         assert inst._flann_target_precision == 0.92
         assert inst._flann_sample_fraction == 0.71
         assert inst._flann_autotune is True
         assert inst._rand_seed == 7
         assert inst._use_sp is True
         assert inst.parallel == 3
Exemple #2
0
        def test_configuration(self, mock_scd):
            default_config = ColorDescriptor_Image_csift.get_default_config()
            default_config['model_directory'] = '/some/path/models/'
            default_config['work_directory'] = '/some/path/work/'

            inst = ColorDescriptor_Image_csift.from_config(default_config)
            ntools.assert_equal(default_config, inst.get_config())
            inst2 = ColorDescriptor_Image_csift.from_config(inst.get_config())
            ntools.assert_equal(inst.get_config(), inst2.get_config())
            ntools.assert_equal(inst._model_dir, inst2._model_dir)
            ntools.assert_equal(inst._work_dir, inst2._work_dir)
            ntools.assert_equal(inst._kmeans_k, inst2._kmeans_k)
            ntools.assert_equal(inst._flann_target_precision,
                                inst2._flann_target_precision)
            ntools.assert_equal(inst._flann_sample_fraction,
                                inst2._flann_sample_fraction)
            ntools.assert_equal(inst._flann_autotune, inst2._flann_autotune)
            ntools.assert_equal(inst._use_sp, inst2._use_sp)
            ntools.assert_equal(inst._rand_seed, inst2._rand_seed)
        def test_configuration(self, _mock_scd):
            default_config = ColorDescriptor_Image_csift.get_default_config()
            default_config['model_directory'] = '/some/path/models/'
            default_config['work_directory'] = '/some/path/work/'

            inst = ColorDescriptor_Image_csift.from_config(default_config)
            self.assertEqual(
                default_config,
                inst.get_config()
            )
            inst2 = ColorDescriptor_Image_csift.from_config(inst.get_config())
            self.assertEqual(inst.get_config(), inst2.get_config())
            self.assertEqual(inst._model_dir, inst2._model_dir)
            self.assertEqual(inst._work_dir, inst2._work_dir)
            self.assertEqual(inst._kmeans_k, inst2._kmeans_k)
            self.assertEqual(inst._flann_target_precision,
                             inst2._flann_target_precision)
            self.assertEqual(inst._flann_sample_fraction,
                             inst2._flann_sample_fraction)
            self.assertEqual(inst._flann_autotune, inst2._flann_autotune)
            self.assertEqual(inst._use_sp, inst2._use_sp)
            self.assertEqual(inst._rand_seed, inst2._rand_seed)
Exemple #4
0
from __future__ import division, print_function
import unittest

import mock
import nose.tools as ntools

from smqtk.algorithms.descriptor_generator import get_descriptor_generator_impls
from smqtk.algorithms.descriptor_generator.colordescriptor.colordescriptor \
    import ColorDescriptor_Image_csift  # arbitrary leaf class

if ColorDescriptor_Image_csift.is_usable():

    class TestColorDescriptor(unittest.TestCase):
        def test_impl_findable(self):
            ntools.assert_in(ColorDescriptor_Image_csift.__name__,
                             get_descriptor_generator_impls())

        @mock.patch('smqtk.algorithms.descriptor_generator'
                    '.colordescriptor.colordescriptor'
                    '.file_utils.safe_create_dir')
        def test_configuration(self, mock_scd):
            default_config = ColorDescriptor_Image_csift.get_default_config()
            default_config['model_directory'] = '/some/path/models/'
            default_config['work_directory'] = '/some/path/work/'

            inst = ColorDescriptor_Image_csift.from_config(default_config)
            ntools.assert_equal(default_config, inst.get_config())
            inst2 = ColorDescriptor_Image_csift.from_config(inst.get_config())
            ntools.assert_equal(inst.get_config(), inst2.get_config())
            ntools.assert_equal(inst._model_dir, inst2._model_dir)
            ntools.assert_equal(inst._work_dir, inst2._work_dir)
from __future__ import division, print_function
import unittest

import mock

from smqtk.algorithms.descriptor_generator import get_descriptor_generator_impls
from smqtk.algorithms.descriptor_generator.colordescriptor.colordescriptor \
    import ColorDescriptor_Image_csift  # arbitrary leaf class


if ColorDescriptor_Image_csift.is_usable():

    class TestColorDescriptor (unittest.TestCase):

        def test_impl_findable(self):
            self.assertIn(ColorDescriptor_Image_csift.__name__,
                          get_descriptor_generator_impls())

        @mock.patch('smqtk.algorithms.descriptor_generator'
                    '.colordescriptor.colordescriptor'
                    '.file_utils.safe_create_dir')
        def test_configuration(self, _mock_scd):
            default_config = ColorDescriptor_Image_csift.get_default_config()
            default_config['model_directory'] = '/some/path/models/'
            default_config['work_directory'] = '/some/path/work/'

            inst = ColorDescriptor_Image_csift.from_config(default_config)
            self.assertEqual(
                default_config,
                inst.get_config()
            )
import unittest

import unittest.mock as mock
import pytest

from smqtk.algorithms.descriptor_generator import DescriptorGenerator
from smqtk.algorithms.descriptor_generator.colordescriptor.colordescriptor \
    import ColorDescriptor_Image_csift  # arbitrary leaf class
from smqtk.utils.configuration import configuration_test_helper


@pytest.mark.skipif(not ColorDescriptor_Image_csift.is_usable(),
                    reason="ColorDescriptor generator is not currently usable")
class TestColorDescriptor(unittest.TestCase):
    def test_impl_findable(self):
        self.assertIn(ColorDescriptor_Image_csift.__name__,
                      DescriptorGenerator.get_impls())

    @mock.patch('smqtk.algorithms.descriptor_generator'
                '.colordescriptor.colordescriptor.safe_create_dir')
    def test_configuration(self, _mock_scd):
        i = ColorDescriptor_Image_csift(
            model_directory='test model dir',
            work_directory='test work dir',
            model_gen_descriptor_limit=123764,
            kmeans_k=42,
            flann_distance_metric='hik',
            flann_target_precision=0.92,
            flann_sample_fraction=0.71,
            flann_autotune=True,
            random_seed=7,