def test_extra_arguments(self, capsys):
'''Test that extra arguments to Computation subclasses are handled
correctly.
'''
env = Environment(ConfigObj())
my_param = 3
env.register_class(self.TestArguments, None, True,
my_param, kwarg=10)
env.run()
out, _ = capsys.readouterr()
assert out == "3\n10\n"
def test_extra_arguments(self, capsys):
'''Test that extra arguments to Computation subclasses are handled
correctly.
'''
env = Environment(ConfigObj())
my_param = 3
env.register_class(self.TestArguments, None, True, my_param, kwarg=10)
env.run()
out, _ = capsys.readouterr()
assert out == "3\n10\n"
def test_running(self, capsys):
'''Test runs with several classes.'''
config = ConfigObj({
'NumPrinter' : {
'number' : 1,
},
})
# No object-specific configurations
env = Environment(config)
for i in range(3):
env.register_class(self.NumPrinter)
env.run()
out, _ = capsys.readouterr()
assert out == "1\n1\n1\n"
# Each object except the first has a specific configuration
env = Environment(config)
env.register_plotter(self.NumPrinter)
for i in range(3):
env.register_class(self.NumPrinter, config={
'NumPrinter': {
'number' : i + 2,
},
})
env.run()
out, _ = capsys.readouterr()
assert out == "1\n2\n3\n4\n"
def test_running(self, capsys):
'''Test runs with several classes.'''
config = ConfigObj({
'NumPrinter': {
'number': 1,
},
})
# No object-specific configurations
env = Environment(config)
for i in range(3):
env.register_class(self.NumPrinter)
env.run()
out, _ = capsys.readouterr()
assert out == "1\n1\n1\n"
# Each object except the first has a specific configuration
env = Environment(config)
env.register_plotter(self.NumPrinter)
for i in range(3):
env.register_class(self.NumPrinter,
config={
'NumPrinter': {
'number': i + 2,
},
})
env.run()
out, _ = capsys.readouterr()
assert out == "1\n2\n3\n4\n"
def test_config(self):
'''Test config correctness.'''
config = ConfigObj({'test': 10})
env = Environment(config)
assert env.config == config