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