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()
)
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)
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,
