def test12(self): global_options = options.GlobalOptions() global_options.read_from_ini_string(self.ini_contents) print(global_options.to_ini_string()) ini_string = global_options.to_ini_string() self.assertTrue(ini_string.find("= a string") > 0)
def test3(self): global_options = options.GlobalOptions() global_options.config.read_file(io.StringIO(self.ini_contents)) instance = OptionsTestsClass() instance.option_sections = ('unknown') option = options.option(OptionsTestsClass.string_option, global_options=global_options) self.assertEqual(option.__get__(instance, type(instance)), "default string") option = options.option(OptionsTestsClass.int_option, type="int", global_options=global_options) self.assertEqual(option.__get__(instance, type(instance)), 2) option = options.option(OptionsTestsClass.boolean_option, type="boolean", global_options=global_options) self.assertEqual(option.__get__(instance, type(instance)), False) option = options.option(OptionsTestsClass.float_option, type="float", global_options=global_options) self.assertEqual(option.__get__(instance, type(instance)), 2.5)
def test9(self): global_options = options.GlobalOptions() global_options.config.read_file(io.StringIO(self.ini_contents)) class A(options.OptionalAttributes): option_sections = ('amuse') def __init__(self, **optional_arguments): options.OptionalAttributes.__init__(self, **optional_arguments) @options.option(type="int", global_options=global_options) def int_option_a(self): return self.i class B(options.OptionalAttributes): option_sections = ('amuse') def __init__(self, **optional_arguments): options.OptionalAttributes.__init__(self, **optional_arguments) @options.option(type="int", global_options=global_options) def int_option_b(self): return 10 class C(A, B): def __init__(self, **optional_arguments): A.__init__(self, **optional_arguments) B.__init__(self, **optional_arguments) instance = C(int_option_a=14) self.assertEqual(instance.int_option_a, 14) self.assertEqual(instance.int_option_b, 10)
def test1(self): global_options = options.GlobalOptions() global_options.config.read_file(io.StringIO(self.ini_contents)) instance = OptionsTestsClass() option = options.option(OptionsTestsClass.string_option, global_options=global_options) self.assertEqual(option.__get__(instance, type(instance)), "a string")
def test5(self): global_options = options.GlobalOptions() global_options.config.read_file(io.StringIO(self.ini_contents)) instance = OptionsTestsClass() instance.option_sections = ('unknown') option = options.option(type="int", global_options=global_options) option = option(OptionsTestsClass.int_option) self.assertEqual(option.__get__(instance, type(instance)), 2)
def test4(self): global_options = options.GlobalOptions() global_options.config.readfp(io.StringIO(self.ini_contents)) instance = OptionsTestsClass() option = options.option(OptionsTestsClass.int_option, type="int", sections=("bhtree", ), global_options=global_options) self.assertEqual(option.__get__(instance, type(instance)), 1) instance.option_sections = ('unknown') self.assertEqual(option.__get__(instance, type(instance)), 3)
def test6(self): global_options = options.GlobalOptions() global_options.config.read_file(io.StringIO(self.ini_contents)) class DecoratedMethods(options.OptionalAttributes): option_sections = ('unknown') @options.option(type="int", global_options=global_options) def int_option(self): return self.i instance = DecoratedMethods() instance.i = 10 self.assertEqual(instance.int_option, 10)
def test10(self): global_options = options.GlobalOptions() self.assertEqual(global_options.rcfilename, 'amuserc') global_options.rcfilename = 'test_for_amuserc' with open(global_options.rcfilename, "w") as f: f.write(self.ini_contents) self.assertTrue(global_options.rcfilepath.endswith('test_for_amuserc')) global_options.load() instance = OptionsTestsClass() option = options.option(OptionsTestsClass.int_option, type="int", global_options=global_options) self.assertEqual(option.__get__(instance, type(instance)), 1) os.remove(global_options.rcfilepath)
def test2(self): global_options = options.GlobalOptions() global_options.config.read_file(io.StringIO(self.ini_contents)) instance = OptionsTestsClass() option = options.option(OptionsTestsClass.int_option, type="int", global_options=global_options) self.assertEqual(option.__get__(instance, type(instance)), 1) option = options.option(OptionsTestsClass.boolean_option, type="boolean", global_options=global_options) self.assertEqual(option.__get__(instance, type(instance)), True) option = options.option(OptionsTestsClass.float_option, type="float", global_options=global_options) self.assertEqual(option.__get__(instance, type(instance)), 1.5)
def test11(self): global_options = options.GlobalOptions() global_options.config.read_file(io.StringIO(self.ini_contents)) class DecoratedMethods(options.OptionalAttributes): option_sections = ('amuse', ) def __init__(self, **optional_arguments): options.OptionalAttributes.__init__(self, **optional_arguments) @options.option(choices=("1", "2"), global_options=global_options) def int_option(self): return 2 instance = DecoratedMethods() self.assertEqual(instance.int_option, "1") all_options = list(instance.iter_options()) self.assertEqual(len(all_options), 1) self.assertEqual(all_options[0].name, "int_option")
def test7(self): global_options = options.GlobalOptions() global_options.config.read_file(io.StringIO(self.ini_contents)) class DecoratedMethods(options.OptionalAttributes): option_sections = ('unknown') i = 8 def __init__(self, **optional_arguments): options.OptionalAttributes.__init__(self, **optional_arguments) @options.option(type="int", global_options=global_options) def int_option(self): return self.i instance = DecoratedMethods() instance.i = 10 instance.int_option = 12 self.assertEqual(instance.int_option, 12) instance = DecoratedMethods(int_option=14) self.assertEqual(instance.int_option, 14)