def test_shortcut_options(self):
trace_actions = get_option('cas.trace_actions')
index_name = get_option('cas.dataset.index_name')
self.assertEqual(get_option('trace_actions'), trace_actions)
self.assertEqual(options.trace_actions, trace_actions)
options.trace_actions = True
self.assertEqual(get_option('cas.trace_actions'), True)
self.assertEqual(options.cas.trace_actions, True)
self.assertEqual(options.trace_actions, True)
self.assertEqual(get_option('index_name'), index_name)
self.assertEqual(get_option('dataset.index_name'), index_name)
self.assertEqual(options.index_name, index_name)
options.index_name = 'Foo'
self.assertEqual(get_option('index_name'), 'Foo')
self.assertEqual(get_option('dataset.index_name'), 'Foo')
self.assertEqual(options.index_name, 'Foo')
reset_option('index_name')
self.assertEqual(get_option('index_name'), '_Index_')
self.assertEqual(get_option('dataset.index_name'), '_Index_')
self.assertEqual(options.index_name, '_Index_')
def _test_method_subscribers(self):
opts = {}
class OptionsSubscriber(object):
def options_subscriber(self, key, value, opts=opts):
opts[key] = value
os = OptionsSubscriber()
num_subscribers = len(_subscribers)
subscribe(os.options_subscriber)
self.assertEqual(len(_subscribers), num_subscribers + 1)
options.cas.print_messages = False
self.assertEqual(opts, {'cas.print_messages': False})
options.cas.print_messages = True
self.assertEqual(opts, {'cas.print_messages': True})
options.cas.dataset.auto_index_style = 'sas'
self.assertEqual(opts, {'cas.print_messages': True,
'cas.dataset.auto_index_style': 'sas'})
options.cas.dataset.auto_index_style = 'pandas'
self.assertEqual(opts, {'cas.print_messages': True,
'cas.dataset.auto_index_style': 'pandas'})
options.cas.print_messages = False
self.assertEqual(opts, {'cas.print_messages': False,
'cas.dataset.auto_index_style': 'pandas'})
reset_option('cas.print_messages')
self.assertEqual(opts, {'cas.print_messages': True,
'cas.dataset.auto_index_style': 'pandas'})
unsubscribe(os.options_subscriber)
self.assertEqual(len(_subscribers), num_subscribers)
subscribe(os.options_subscriber)
self.assertEqual(len(_subscribers), num_subscribers + 1)
del os.options_subscriber
self.assertEqual(len(_subscribers), num_subscribers)
options.cas.print_messages = False
self.assertEqual(opts, {'cas.print_messages': True,
'cas.dataset.auto_index_style': 'pandas'})
def test_function_subscribers(self):
opts = {}
def options_subscriber(key, value, opts=opts):
opts[key] = value
num_subscribers = len(_subscribers)
subscribe(options_subscriber)
self.assertEqual(len(_subscribers), num_subscribers + 1)
options.cas.print_messages = True
self.assertEqual(opts, {'cas.print_messages': True})
options.cas.print_messages = False
self.assertEqual(opts, {'cas.print_messages': False})
options.cas.dataset.index_name = 'foo'
self.assertEqual(opts, {'cas.print_messages': False,
'cas.dataset.index_name': 'foo'})
options.cas.dataset.index_name = 'bar'
self.assertEqual(opts, {'cas.print_messages': False,
'cas.dataset.index_name': 'bar'})
options.cas.print_messages = True
self.assertEqual(opts, {'cas.print_messages': True,
'cas.dataset.index_name': 'bar'})
options.cas.print_messages = False
reset_option('cas.print_messages')
self.assertEqual(opts, {'cas.print_messages': True,
'cas.dataset.index_name': 'bar'})
unsubscribe(options_subscriber)
self.assertEqual(len(_subscribers), num_subscribers)
subscribe(options_subscriber)
self.assertEqual(len(_subscribers), num_subscribers + 1)
del options_subscriber
self.assertEqual(len(_subscribers), num_subscribers)
options.cas.print_messages = False
self.assertEqual(opts, {'cas.print_messages': True,
'cas.dataset.index_name': 'bar'})
def test_basic(self):
self.assertEqual(get_option('cas.print_messages'), True)
# self.assertEqual(get_option('display.notebook.repr_html'), True)
# self.assertEqual(get_option('display.notebook.repr_javascript'), False)
set_option('cas.print_messages', False)
self.assertEqual(get_option('cas.print_messages'), False)
with self.assertRaises(SWATOptionError):
options.cas.print_messages = 'foo'
options.cas.print_messages = True
self.assertEqual(options.cas.print_messages, True)
with self.assertRaises(SWATOptionError):
options.cas.print_messages = 10
self.assertEqual(options.cas.print_messages, True)
self.assertEqual(type(options.cas), type(options))
# This key exists, but it's a level in the hierarchy, not an option
# with self.assertRaises(SWATOptionError):
# get_option('display.notebook.css')
options.cas.print_messages = False
reset_option('cas.print_messages')
self.assertEqual(options.cas.print_messages, True)
with self.assertRaises(SWATOptionError):
reset_option('cas.foo')
with self.assertRaises(SWATOptionError):
reset_option('cas')
def tearDown(self):
reset_option()
def setUp(self):
reset_option()
