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)
