def test_tracebacks(self):
'Tracebacks have correct file names and line numbers'
# If we want tracebacks to make sense, then the reported file names and
# line numbers need to associate with the code being executed
# regardless which block represents and executes the code.
def test(tb, lineno, filename):
self.assertEqual(tb.tb_lineno, lineno+1)
self.assertEqual(tb.tb_frame.f_code.co_filename, filename)
def tracebacks():
"A list of the current exception's traceback objects."
tb = sys.exc_info()[2]
l = [tb]
while tb.tb_next is not None:
tb = tb.tb_next
l.append(tb)
return l
class File(StringIO, object):
"Extend StringIO with a 'name' attribute."
def __init__(self, name, *args, **kw):
super(File, self).__init__(*args, **kw)
self.name = name
a = Block(File('foo/a.py', 'y = x'))
try:
a.execute({})
except NameError, e:
test(tracebacks()[-1], 1, 'foo/a.py')
def test_basic_02():
"""Another test of the basic use of a Block."""
code = 'y = x + 1'
b = Block(code)
assert_equal(b.inputs, set(['x']))
assert_equal(b.outputs, set(['y']))
names = dict(x=100)
b.execute(names)
assert_equal(sorted(names), ['x', 'y'])
assert_equal(names['x'], 100)
assert_equal(names['y'], 101)
def test_basic_01():
"""Test basic use of a Block."""
code = 'x = 100\ny = x + 1'
b = Block(code)
assert_equal(b.inputs, set([]))
assert_equal(b.outputs, set(['x','y']))
names = dict()
b.execute(names)
assert_equal(sorted(names), ['x', 'y'])
assert_equal(names['x'], 100)
assert_equal(names['y'], 101)
def test_basic_02():
"""Another test of the basic use of a Block."""
code = 'y = x + 1'
b = Block(code)
assert_equal(b.inputs, set(['x']))
assert_equal(b.outputs, set(['y']))
names = dict(x=100)
b.execute(names)
assert_equal(sorted(names), ['x', 'y'])
assert_equal(names['x'], 100)
assert_equal(names['y'], 101)
def test_basic_01():
"""Test basic use of a Block."""
code = 'x = 100\ny = x + 1'
b = Block(code)
assert_equal(b.inputs, set([]))
assert_equal(b.outputs, set(['x', 'y']))
assert_equal(b.const_assign, (set(['x']), ['x = 100']))
names = dict()
b.execute(names)
assert_equal(sorted(names), ['x', 'y'])
assert_equal(names['x'], 100)
assert_equal(names['y'], 101)
def test_optimization_no_filenames_in_tracebacks(self):
'Optimization: No filenames in tracebacks'
b = Block('import operator\n'
'x,d = 2,{}\n'
'for i in range(10):\n'
' d[i] = repr(i)\n'
' x *= 2\n'
'sum = lambda l, add=operator.add: reduce(add, l)\n'
'd = (sum(d.keys()), sum(d.values()))\n',
no_filenames_in_tracebacks = True)
c = {}
b.execute(c)
assert c['x'] == 2**11
assert c['d'] == (9*10/2, '0123456789')
def test_caching_code(self):
"Caching: '_code'"
b,c = Block('a=2'), {}
b._code
b.ast = Block('a=3').ast
b.execute(c)
self.assertEqual(c['a'], 3)
b,c = Block('a=2'), {}
b._code
b.sub_blocks = [Block('a=3')]
b.execute(c)
self.assertEqual(c['a'], 3)
b,c = Block('a=3; a=2'), {}
b._code
b.sub_blocks.pop()
b.execute(c)
self.assertEqual(c['a'], 3)
b,c = Block(''), {}
b._code
b.sub_blocks.append(Block('a=3'))
b.sub_blocks = [Block('a=3')]
b.execute(c)
assert 'a' in c
def test_caching_code(self):
"Caching: '_code'"
b, c = Block('a=2'), {}
b._code
b.ast = Block('a=3').ast
b.execute(c)
self.assertEqual(c['a'], 3)
b, c = Block('a=2'), {}
b._code
b.sub_blocks = [Block('a=3')]
b.execute(c)
self.assertEqual(c['a'], 3)
b, c = Block('a=3; a=2'), {}
b._code
b.sub_blocks.pop()
b.execute(c)
self.assertEqual(c['a'], 3)
b, c = Block(''), {}
b._code
b.sub_blocks.append(Block('a=3'))
b.sub_blocks = [Block('a=3')]
b.execute(c)
assert 'a' in c
def test3(self):
""" Same sized array assignments within 'with' block
"""
file_object = open(os.path.join(self.code_dir, 'test3.py'), 'r')
code = file_object.read()
file_object.close()
b = Block(code)
b.execute(self.context)
desired_vp = arange(0., 10., 1.)
desired_vs = arange(0., 100., 10.)
# Check equal
self.assertTrue((desired_vp == self.context['vp']).all())
self.assertTrue((desired_vs == self.context['vs']).all())
def test3(self):
""" Same sized array assignments within 'with' block
"""
file_object = open(os.path.join(self.code_dir,'test3.py'), 'r')
code = file_object.read()
file_object.close()
b = Block(code)
b.execute(self.context)
desired_vp = arange(0., 10., 1.)
desired_vs = arange(0., 100., 10.)
# Check equal
self.assertTrue((desired_vp == self.context['vp']).all())
self.assertTrue((desired_vs == self.context['vs']).all())
def test2(self):
""" Different sized arrays
"""
file_object = open(os.path.join(self.code_dir, 'test2.py'), 'r')
code = file_object.read()
file_object.close()
b = Block(code)
b.execute(self.context)
depth = arange(0., 10000., 1000.)
desired_vp = zeros(depth.shape)
desired_vs = zeros(depth.shape)
desired_vp[(depth < 4000.0) & (depth > 1000.0)] = 1.0
desired_vs[(depth < 4000.0) & (depth > 1000.0)] = 1.5
# Check equal
self.assertTrue((desired_vp == self.context['vp']).all())
self.assertTrue((desired_vs == self.context['vs']).all())
def test2(self):
""" Different sized arrays
"""
file_object = open(os.path.join(self.code_dir,'test2.py'), 'r')
code = file_object.read()
file_object.close()
b = Block(code)
b.execute(self.context)
depth = arange(0., 10000., 1000.)
desired_vp = zeros(depth.shape)
desired_vs = zeros(depth.shape)
desired_vp[(depth<4000.0) & (depth > 1000.0)] = 1.0
desired_vs[(depth<4000.0) & (depth > 1000.0)] = 1.5
# Check equal
self.assertTrue((desired_vp == self.context['vp']).all())
self.assertTrue((desired_vs == self.context['vs']).all())
def test_block_events(self):
import numpy
from codetools.blocks.api import Block
context = DataContext(name="data")
context.on_trait_change(self.event_listener, 'items_modified')
context.defer_events = True
context['a'] = 4
context['b'] = numpy.array((1,2,3))
context.defer_events = False
self.assertEqual(self.event_count, 1)
multi_context = MultiContext(context, name="multi")
multi_context.on_trait_change(self.event_listener, 'items_modified')
block = Block("c = a * b")
block.execute(multi_context)
# we expect one event from data context, one from multi context
self.assertEqual(self.event_count, 3)
def test_block_events(self):
import numpy
from codetools.blocks.api import Block
context = DataContext(name="data")
context.on_trait_change(self.event_listener, 'items_modified')
context.defer_events = True
context['a'] = 4
context['b'] = numpy.array((1, 2, 3))
context.defer_events = False
self.assertEqual(self.event_count, 1)
multi_context = MultiContext(context, name="multi")
multi_context.on_trait_change(self.event_listener, 'items_modified')
block = Block("c = a * b")
block.execute(multi_context)
# we expect one event from data context, one from multi context
self.assertEqual(self.event_count, 3)
class ParametricInteractorTestCase(unittest.TestCase):
""" Unit testing for ParametricInteractor
"""
def setUp(self):
code = "from blockcanvas.debug.my_operator import add, mul\n" \
"c = add(a,b)\n" \
"d = mul(c, 2)\n" \
"e = mul(c, 3)\n" \
"f = add(d,e)"
self.block=Block(code)
# Context setup.
self.context = MultiContext(DataContext(name='Data'),{})
self.context['a'] = 1
self.context['b'] = 2
def test_attributes(self):
""" Test if creation of attributes is working correctly
"""
interactor = ParametricInteractor(context=self.context,block=self.block)
self.assertTrue(hasattr(interactor, interactor._input_prefix + "a"))
self.assertTrue(hasattr(interactor, interactor._input_prefix + "b"))
# Check if the parameter items were written correctly
attribute_a = getattr(interactor, interactor._input_prefix+'a')
attribute_b = getattr(interactor, interactor._input_prefix+'b')
self.assertEqual(attribute_a.step, 0)
self.assertEqual(attribute_a.low, 1)
self.assertEqual(attribute_a.high, 1)
self.assertEqual(attribute_b.step, 0)
self.assertEqual(attribute_b.low, 2)
self.assertEqual(attribute_b.high, 2)
# Check if the inputs in the interactor are right
self.assertEqual(interactor.inputs, list(self.block.inputs))
def test_create_shadows(self):
""" Test if creating shadows is working correctly
"""
interactor = ParametricInteractor(context=self.context,block=self.block)
# Change the parameters for the attributes
attribute_a = getattr(interactor, interactor._input_prefix+'a')
attribute_b = getattr(interactor, interactor._input_prefix+'b')
attribute_a.high = 4
attribute_a.step = 1
attribute_b.high = 3
attribute_b.step = 1
# Create the shadows.
interactor._update_contexts_button_changed()
# Check if the shadows were created correctly.
self.assertEqual(len(self.context.shadows),
len(attribute_a.output_list)*len(attribute_b.output_list))
# Check if the shadow context gives desired result.
self.block.execute(self.context.shadows[-1])
self.assertEqual(self.context.shadows[-1]['f'],
5*(self.context.shadows[-1]['a']+self.context.shadows[-1]['b']))
class ConfigurableInteractorTestCase(unittest.TestCase):
""" Unit testing for ConfigurableInteractor
"""
def setUp(self):
code = "from blockcanvas.debug.my_operator import add,mul\n"\
"c = add(a, b)\n" \
"d = mul(c, 2)\n" \
"e = mul(c, 3)\n" \
"f = add(d, e)\n" \
"g = add(f, 3)\n" \
"h = mul(g, f)\n" \
"i = mul(g,h)"
self.block=Block(code)
# Context setup.
self.context = MultiContext(DataContext(name='Data'),{})
self.context['a'] = 1.0
self.context['b'] = 2
self.block.execute(self.context)
# ConfigurableInteractor setup
i_config = InteractorConfig(var_configs=[VariableConfig(name='a',
type='Shadow'),
VariableConfig(name='b',
type='Parametric'),
VariableConfig(name='c',
type='Shadow'),
VariableConfig(name='d',
type='Parametric'),
VariableConfig(name='e',
type='Stochastic: Constant'),
VariableConfig(name='f',
type='Stochastic: Gaussian'),
VariableConfig(name='g',
type='Stochastic: Triangular'),
VariableConfig(name='h',
type='Stochastic: Uniform'),
])
self.interactor = ConfigurableInteractor(interactor_config=i_config,
block=self.block,
context=self.context)
# Temp dir
self.temp_dir = tempfile.gettempdir()
def test_build_vars(self):
""" Does the interactor create the right traits for the types of
of interactors
"""
self.failUnless(isinstance(self.interactor.interactor_shadow,
ShadowInteractor))
self.failUnless(isinstance(self.interactor.interactor_parametric,
ParametricInteractor))
self.failUnless(isinstance(
self.interactor.interactor_stochastic_constant,StochasticInteractor))
self.failUnless(isinstance(
self.interactor.interactor_stochastic_gaussian,StochasticInteractor))
self.failUnless(isinstance(
self.interactor.interactor_stochastic_triangular,
StochasticInteractor))
self.failUnless(isinstance(
self.interactor.interactor_stochastic_uniform, StochasticInteractor))
def test_load_project_ranges_for_shadow_interactor(self):
""" Does loading ranges from file work for shadow interactor ?
"""
child_int = self.interactor.interactor_shadow
file_path = os.path.join(os.path.dirname(__file__), 'data',
'project_ranges.txt')
# Applying the ranges to the shadow interactor in the configurable
# interactor
self.interactor.load_ranges_from_files({'shadow':file_path})
child_int._view_items()
self.assertEqual(len(child_int.ranges.keys()), 42)
self.assertAlmostEqual(child_int.input_a__high, 5)
self.assertAlmostEqual(child_int.input_a__low, 3.4)
self.assertEqual(child_int.input_a, self.context['a'])
def test_save_project_ranges_for_shadow_interactor(self):
""" Does saving ranges to file work for shadow interactor ?
"""
child_int = self.interactor.interactor_shadow
f_path = os.path.join(self.temp_dir,'test_save_ranges_for_shadow.txt')
child_int._view_items()
child_int.input_a__high = 5.0
self.interactor.save_ranges_to_files({'shadow': f_path})
# Read the saved file
file_object = open(f_path, 'r')
line = file_object.read()
line = [name.strip() for name in line.split()]
file_object.close()
os.remove(f_path)
# Check results
self.assertEqual(len(line), 6)
self.assertEqual(line[2], str(5.0))
def test_load_project_ranges_for_parametric_interactor(self):
""" Does loading ranges from file work for parametric interactor ?
"""
child_int = self.interactor.interactor_parametric
file_path = os.path.join(os.path.dirname(__file__), 'data',
'project_ranges.txt')
self.interactor.load_ranges_from_files({'parametric': file_path})
child_int._view_items()
self.assertEqual(len(child_int.ranges.keys()), 42)
self.assertAlmostEqual(child_int.input_b.low, 1.0)
self.assertAlmostEqual(child_int.input_b.high, 6.0)
self.assertEqual(child_int.input_b.step, 0.0)
self.assertEqual(child_int.input_b.input_value, self.context['b'])
def test_save_project_ranges_for_parametric_interactor(self):
""" Does saving ranges to file work for parametric interactor ?
"""
file_path = os.path.join(self.temp_dir,
'test_save_ranges_for_parametric.txt')
child_int = self.interactor.interactor_parametric
child_int._view_items()
child_int.input_b.low = 2
child_int.input_b.high = 3
child_int.input_b.step = 0.5
self.interactor.save_ranges_to_files({'parametric':file_path})
# Read the saved file
file_object = open(file_path, 'r')
lines = file_object.readlines()
lines = [line.split() for line in lines]
file_object.close()
os.remove(file_path)
self.assertEqual(len(lines), 2)
self.assertEqual(len(lines[0]), 4)
self.assertEqual(len(lines[1]), 4)
self.assertEqual(lines[0][1], str(2))
self.assertEqual(lines[0][2], str(3))
self.assertEqual(lines[0][3], str(0))
def test_load_project_ranges_for_stochastic_constant_interactor(self):
""" Does loading parameters from file work for stochastic interactor
with constant distribution ?
"""
child_int = self.interactor.interactor_stochastic_constant
file_path = os.path.join(os.path.dirname(__file__), 'data',
'stochastic_constant_ranges.txt')
self.interactor.load_ranges_from_files({'stochastic_constant':
file_path})
child_int._view_items()
self.assertEqual(child_int.input_e.distribution.value, 45)
def test_save_project_ranges_for_stochastic_constant_interactor(self):
""" Does saving parameters to file work for stochastic interactor with
constant distribution ?
"""
file_path = os.path.join(self.temp_dir,
'test_save_ranges_for_stoch_constant.txt')
child_int = self.interactor.interactor_stochastic_constant
child_int._view_items()
child_int.input_e.distribution.value = 37.5
self.interactor.save_ranges_to_files({'stochastic_constant': file_path})
# Read the saved file
file_object = open(file_path, 'r')
lines = file_object.readlines()
lines = [line.split() for line in lines]
file_object.close()
os.remove(file_path)
self.assertEqual(len(lines), 1)
self.assertEqual(len(lines[0]), 2)
self.assertEqual(lines[0][1], str(37.5))
def test_load_project_ranges_for_stochastic_gaussian_interactor(self):
""" Does loading parameters from file work for stochastic interactor
with gaussian distribution ?
"""
child_int = self.interactor.interactor_stochastic_gaussian
file_path = os.path.join(os.path.dirname(__file__), 'data',
'stochastic_gaussian_ranges.txt')
self.interactor.load_ranges_from_files({'stochastic_gaussian':
file_path})
child_int._view_items()
self.assertEqual(child_int.input_f.distribution.mean, 39)
self.assertEqual(child_int.input_f.distribution.std, 3)
def test_save_project_ranges_for_stochastic_gaussian_interactor(self):
""" Does saving parameters to file work for stochastic interactor
with gaussian distribution ?
"""
file_path = os.path.join(self.temp_dir,
'test_save_ranges_for_stoch_gaussian.txt')
child_int = self.interactor.interactor_stochastic_gaussian
child_int._view_items()
child_int.input_f.distribution.mean = 200
child_int.input_f.distribution.std = 0.5
self.interactor.save_ranges_to_files({'stochastic_gaussian': file_path})
# Read the saved file
file_object = open(file_path, 'r')
lines = file_object.readlines()
lines = [line.split() for line in lines]
file_object.close()
os.remove(file_path)
self.assertEqual(len(lines), 1)
self.assertEqual(len(lines[0]), 3)
self.assertEqual(lines[0][1], str(200.0))
self.assertEqual(lines[0][2], str(0.5))
def test_load_project_ranges_for_stochastic_triangular_interactor(self):
""" Does loading parameters from file work for triangular interactor
with triangular distribution ?
"""
child_int = self.interactor.interactor_stochastic_triangular
file_path = os.path.join(os.path.dirname(__file__), 'data',
'stochastic_triangular_ranges.txt')
self.interactor.load_ranges_from_files({'stochastic_triangular':
file_path})
child_int._view_items()
self.assertEqual(child_int.input_g.distribution.mode, 44)
self.assertEqual(child_int.input_g.distribution.low, 39)
self.assertEqual(child_int.input_g.distribution.high, 49)
def test_save_project_ranges_for_stochastic_triangular_interactor(self):
""" Does saving parameters to file work for stochastic interactor
with triangular distribution ?
"""
file_path = os.path.join(self.temp_dir,
'test_save_ranges_for_stoch_triangular.txt')
child_int = self.interactor.interactor_stochastic_triangular
child_int._view_items()
child_int.input_g.distribution.mode = 40.0
child_int.input_g.distribution.low = 30.0
child_int.input_g.distribution.high = 50.0
self.interactor.save_ranges_to_files({'stochastic_triangular':
file_path})
# Read the saved file
file_object = open(file_path, 'r')
lines = file_object.readlines()
lines = [line.split() for line in lines]
file_object.close()
os.remove(file_path)
self.assertEqual(len(lines), 1)
self.assertEqual(len(lines[0]), 4)
self.assertEqual(lines[0][1], str(40.0))
self.assertEqual(lines[0][2], str(30.0))
self.assertEqual(lines[0][3], str(50.0))
def test_load_project_ranges_for_stochastic_uniform_interactor(self):
""" Does loading parameters from file work for stochastic interactor
with uniform distribution ?
"""
child_int = self.interactor.interactor_stochastic_uniform
file_path = os.path.join(os.path.dirname(__file__), 'data',
'stochastic_uniform_ranges.txt')
self.interactor.load_ranges_from_files({'stochastic_uniform':file_path})
child_int._view_items()
self.assertEqual(child_int.input_h.distribution.low, 40)
self.assertEqual(child_int.input_h.distribution.high, 58)
def test_save_project_ranges_for_stochastic_uniform_interactor(self):
""" Does saving parameters to file work for stochastic interactor
with uniform distribution ?
"""
file_path = os.path.join(self.temp_dir,
'test_save_ranges_for_stoch_uniform.txt')
child_int = self.interactor.interactor_stochastic_uniform
child_int._view_items()
child_int.input_h.distribution.low = 30.0
child_int.input_h.distribution.high = 50.0
self.interactor.save_ranges_to_files({'stochastic_uniform': file_path})
# Read the saved file
file_object = open(file_path, 'r')
lines = file_object.readlines()
lines = [line.split() for line in lines]
file_object.close()
os.remove(file_path)
self.assertEqual(len(lines), 1)
self.assertEqual(len(lines[0]), 3)
self.assertEqual(lines[0][1], str(30.0))
self.assertEqual(lines[0][2], str(50.0))
def test_load_global_ranges_with_project_ranges(self):
""" Do global ranges get overwritten by the project settings ?
"""
code = "from blockcanvas.debug.my_operator import add,mul\n"\
"d = mul(a, b)\n" \
"e = mul(c, 3)\n" \
"f = add(d,e)"
self.block = Block(code)
self.context['c'] = 2.0
# This is to be done because the interactor doesn't get updated with
# change in block.
i_config = InteractorConfig(var_configs=[VariableConfig(name='a',
type='Shadow'),
VariableConfig(name='b',
type='Parametric'),
VariableConfig(name='c',
type='Shadow'),
VariableConfig(name='d',
type='Parametric')])
self.interactor = ConfigurableInteractor(block=self.block,
context=self.context,
interactor_config=i_config)
child_int = self.interactor.interactor_shadow
global_file_path = os.path.join(os.path.dirname(__file__), 'data',
'global_user_range.txt')
project_file_path = os.path.join(os.path.dirname(__file__), 'data',
'project_ranges.txt')
self.interactor.load_ranges_from_files({'shadow':project_file_path},
{'shadow':global_file_path})
child_int._view_items()
# Range of a is obtained from the project settings and not the
# global preferences
self.assertAlmostEqual(child_int.input_a__low, 3.4)
self.assertAlmostEqual(child_int.input_a__high, 5.0)
self.assertNotEqual(child_int.input_a__low, 0.15)
self.assertNotEqual(child_int.input_a__high, 9.3)
# Range of c is obtained from the global preferences
self.assertAlmostEqual(child_int.input_c__low, 2.0)
self.assertAlmostEqual(child_int.input_c__high, 7.0)
def test_load_global_ranges_without_project_ranges(self):
""" Do global ranges get used when there are no project settings ?
"""
code = "from blockcanvas.debug.my_operator import add,mul\n"\
"d = mul(a, b)\n" \
"e = mul(c, 3)\n" \
"f = add(d,e)"
self.block = Block(code)
self.context['c'] = 2.0
# This is to be done because the interactor doesn't get updated with
# change in block.
i_config = InteractorConfig(var_configs=[VariableConfig(name='a',
type='Shadow'),
VariableConfig(name='b',
type='Parametric'),
VariableConfig(name='c',
type='Shadow'),
VariableConfig(name='d',
type='Parametric')])
self.interactor = ConfigurableInteractor(block=self.block,
context=self.context,
interactor_config=i_config)
child_int = self.interactor.interactor_shadow
global_file_path = os.path.join(os.path.dirname(__file__), 'data',
'global_user_range.txt')
self.interactor.load_ranges_from_files({}, {'shadow':global_file_path})
child_int._view_items()
# Range of a is obtained from the project settings and not the
# global preferences
self.assertAlmostEqual(child_int.input_a__low, 0.15)
self.assertAlmostEqual(child_int.input_a__high, 9.3)
# Range of c is obtained from the global preferences
self.assertAlmostEqual(child_int.input_c__low, 2.0)
self.assertAlmostEqual(child_int.input_c__high, 7.0)
'from numpy import zeros\n'\
'from codetools.contexts.with_mask import Mask\n'\
'with Mask((depth < 4000.0) & (depth > 1000.0)):vp=1.5 ; vs=1.0'
## # Expanded form of with statement taken from PEP 343. This is just for testing
## code = 'from numpy import zeros\n'\
## 'array_len = depth.shape\n'\
## 'vp = zeros(array_len)\n'\
## 'vs = zeros(array_len)\n'\
## 'from codetools.contexts.with_mask import Mask\n'\
## 'mgr = Mask((depth < 4000.0) & (depth > 1000.0))\n'\
## 'exit_code = mgr.__exit__\n'\
## 'mgr.__enter__()\n'\
## 'exc = True\n'\
## 'try:\n'\
## ' try:\n'\
## ' vp = 1.0\n'\
## ' vs = 1.5\n'\
## ' except:\n'\
## ' exc = False\n'\
## 'finally:\n'\
## ' if exc:\n'\
## ' exit_code(None, None, None)'
b = Block(code)
b.execute(context)
print('vp', context['vp'])
print('vs', context['vs'])
### EOF ------------------------------------------------------------------------
class ConfigurableInteractorTestCase(unittest.TestCase):
""" Unit testing for ConfigurableInteractor
"""
def setUp(self):
code = "from blockcanvas.debug.my_operator import add,mul\n"\
"c = add(a, b)\n" \
"d = mul(c, 2)\n" \
"e = mul(c, 3)\n" \
"f = add(d, e)\n" \
"g = add(f, 3)\n" \
"h = mul(g, f)\n" \
"i = mul(g,h)"
self.block = Block(code)
# Context setup.
self.context = MultiContext(DataContext(name='Data'), {})
self.context['a'] = 1.0
self.context['b'] = 2
self.block.execute(self.context)
# ConfigurableInteractor setup
i_config = InteractorConfig(var_configs=[
VariableConfig(name='a', type='Shadow'),
VariableConfig(name='b', type='Parametric'),
VariableConfig(name='c', type='Shadow'),
VariableConfig(name='d', type='Parametric'),
VariableConfig(name='e', type='Stochastic: Constant'),
VariableConfig(name='f', type='Stochastic: Gaussian'),
VariableConfig(name='g', type='Stochastic: Triangular'),
VariableConfig(name='h', type='Stochastic: Uniform'),
])
self.interactor = ConfigurableInteractor(interactor_config=i_config,
block=self.block,
context=self.context)
# Temp dir
self.temp_dir = tempfile.gettempdir()
def test_build_vars(self):
""" Does the interactor create the right traits for the types of
of interactors
"""
self.failUnless(
isinstance(self.interactor.interactor_shadow, ShadowInteractor))
self.failUnless(
isinstance(self.interactor.interactor_parametric,
ParametricInteractor))
self.failUnless(
isinstance(self.interactor.interactor_stochastic_constant,
StochasticInteractor))
self.failUnless(
isinstance(self.interactor.interactor_stochastic_gaussian,
StochasticInteractor))
self.failUnless(
isinstance(self.interactor.interactor_stochastic_triangular,
StochasticInteractor))
self.failUnless(
isinstance(self.interactor.interactor_stochastic_uniform,
StochasticInteractor))
def test_load_project_ranges_for_shadow_interactor(self):
""" Does loading ranges from file work for shadow interactor ?
"""
child_int = self.interactor.interactor_shadow
file_path = os.path.join(os.path.dirname(__file__), 'data',
'project_ranges.txt')
# Applying the ranges to the shadow interactor in the configurable
# interactor
self.interactor.load_ranges_from_files({'shadow': file_path})
child_int._view_items()
self.assertEqual(len(child_int.ranges.keys()), 42)
self.assertAlmostEqual(child_int.input_a__high, 5)
self.assertAlmostEqual(child_int.input_a__low, 3.4)
self.assertEqual(child_int.input_a, self.context['a'])
def test_save_project_ranges_for_shadow_interactor(self):
""" Does saving ranges to file work for shadow interactor ?
"""
child_int = self.interactor.interactor_shadow
f_path = os.path.join(self.temp_dir, 'test_save_ranges_for_shadow.txt')
child_int._view_items()
child_int.input_a__high = 5.0
self.interactor.save_ranges_to_files({'shadow': f_path})
# Read the saved file
file_object = open(f_path, 'r')
line = file_object.read()
line = [name.strip() for name in line.split()]
file_object.close()
os.remove(f_path)
# Check results
self.assertEqual(len(line), 6)
self.assertEqual(line[2], str(5.0))
def test_load_project_ranges_for_parametric_interactor(self):
""" Does loading ranges from file work for parametric interactor ?
"""
child_int = self.interactor.interactor_parametric
file_path = os.path.join(os.path.dirname(__file__), 'data',
'project_ranges.txt')
self.interactor.load_ranges_from_files({'parametric': file_path})
child_int._view_items()
self.assertEqual(len(child_int.ranges.keys()), 42)
self.assertAlmostEqual(child_int.input_b.low, 1.0)
self.assertAlmostEqual(child_int.input_b.high, 6.0)
self.assertEqual(child_int.input_b.step, 0.0)
self.assertEqual(child_int.input_b.input_value, self.context['b'])
def test_save_project_ranges_for_parametric_interactor(self):
""" Does saving ranges to file work for parametric interactor ?
"""
file_path = os.path.join(self.temp_dir,
'test_save_ranges_for_parametric.txt')
child_int = self.interactor.interactor_parametric
child_int._view_items()
child_int.input_b.low = 2
child_int.input_b.high = 3
child_int.input_b.step = 0.5
self.interactor.save_ranges_to_files({'parametric': file_path})
# Read the saved file
file_object = open(file_path, 'r')
lines = file_object.readlines()
lines = [line.split() for line in lines]
file_object.close()
os.remove(file_path)
self.assertEqual(len(lines), 2)
self.assertEqual(len(lines[0]), 4)
self.assertEqual(len(lines[1]), 4)
self.assertEqual(lines[0][1], str(2))
self.assertEqual(lines[0][2], str(3))
self.assertEqual(lines[0][3], str(0))
def test_load_project_ranges_for_stochastic_constant_interactor(self):
""" Does loading parameters from file work for stochastic interactor
with constant distribution ?
"""
child_int = self.interactor.interactor_stochastic_constant
file_path = os.path.join(os.path.dirname(__file__), 'data',
'stochastic_constant_ranges.txt')
self.interactor.load_ranges_from_files(
{'stochastic_constant': file_path})
child_int._view_items()
self.assertEqual(child_int.input_e.distribution.value, 45)
def test_save_project_ranges_for_stochastic_constant_interactor(self):
""" Does saving parameters to file work for stochastic interactor with
constant distribution ?
"""
file_path = os.path.join(self.temp_dir,
'test_save_ranges_for_stoch_constant.txt')
child_int = self.interactor.interactor_stochastic_constant
child_int._view_items()
child_int.input_e.distribution.value = 37.5
self.interactor.save_ranges_to_files(
{'stochastic_constant': file_path})
# Read the saved file
file_object = open(file_path, 'r')
lines = file_object.readlines()
lines = [line.split() for line in lines]
file_object.close()
os.remove(file_path)
self.assertEqual(len(lines), 1)
self.assertEqual(len(lines[0]), 2)
self.assertEqual(lines[0][1], str(37.5))
def test_load_project_ranges_for_stochastic_gaussian_interactor(self):
""" Does loading parameters from file work for stochastic interactor
with gaussian distribution ?
"""
child_int = self.interactor.interactor_stochastic_gaussian
file_path = os.path.join(os.path.dirname(__file__), 'data',
'stochastic_gaussian_ranges.txt')
self.interactor.load_ranges_from_files(
{'stochastic_gaussian': file_path})
child_int._view_items()
self.assertEqual(child_int.input_f.distribution.mean, 39)
self.assertEqual(child_int.input_f.distribution.std, 3)
def test_save_project_ranges_for_stochastic_gaussian_interactor(self):
""" Does saving parameters to file work for stochastic interactor
with gaussian distribution ?
"""
file_path = os.path.join(self.temp_dir,
'test_save_ranges_for_stoch_gaussian.txt')
child_int = self.interactor.interactor_stochastic_gaussian
child_int._view_items()
child_int.input_f.distribution.mean = 200
child_int.input_f.distribution.std = 0.5
self.interactor.save_ranges_to_files(
{'stochastic_gaussian': file_path})
# Read the saved file
file_object = open(file_path, 'r')
lines = file_object.readlines()
lines = [line.split() for line in lines]
file_object.close()
os.remove(file_path)
self.assertEqual(len(lines), 1)
self.assertEqual(len(lines[0]), 3)
self.assertEqual(lines[0][1], str(200.0))
self.assertEqual(lines[0][2], str(0.5))
def test_load_project_ranges_for_stochastic_triangular_interactor(self):
""" Does loading parameters from file work for triangular interactor
with triangular distribution ?
"""
child_int = self.interactor.interactor_stochastic_triangular
file_path = os.path.join(os.path.dirname(__file__), 'data',
'stochastic_triangular_ranges.txt')
self.interactor.load_ranges_from_files(
{'stochastic_triangular': file_path})
child_int._view_items()
self.assertEqual(child_int.input_g.distribution.mode, 44)
self.assertEqual(child_int.input_g.distribution.low, 39)
self.assertEqual(child_int.input_g.distribution.high, 49)
def test_save_project_ranges_for_stochastic_triangular_interactor(self):
""" Does saving parameters to file work for stochastic interactor
with triangular distribution ?
"""
file_path = os.path.join(self.temp_dir,
'test_save_ranges_for_stoch_triangular.txt')
child_int = self.interactor.interactor_stochastic_triangular
child_int._view_items()
child_int.input_g.distribution.mode = 40.0
child_int.input_g.distribution.low = 30.0
child_int.input_g.distribution.high = 50.0
self.interactor.save_ranges_to_files(
{'stochastic_triangular': file_path})
# Read the saved file
file_object = open(file_path, 'r')
lines = file_object.readlines()
lines = [line.split() for line in lines]
file_object.close()
os.remove(file_path)
self.assertEqual(len(lines), 1)
self.assertEqual(len(lines[0]), 4)
self.assertEqual(lines[0][1], str(40.0))
self.assertEqual(lines[0][2], str(30.0))
self.assertEqual(lines[0][3], str(50.0))
def test_load_project_ranges_for_stochastic_uniform_interactor(self):
""" Does loading parameters from file work for stochastic interactor
with uniform distribution ?
"""
child_int = self.interactor.interactor_stochastic_uniform
file_path = os.path.join(os.path.dirname(__file__), 'data',
'stochastic_uniform_ranges.txt')
self.interactor.load_ranges_from_files(
{'stochastic_uniform': file_path})
child_int._view_items()
self.assertEqual(child_int.input_h.distribution.low, 40)
self.assertEqual(child_int.input_h.distribution.high, 58)
def test_save_project_ranges_for_stochastic_uniform_interactor(self):
""" Does saving parameters to file work for stochastic interactor
with uniform distribution ?
"""
file_path = os.path.join(self.temp_dir,
'test_save_ranges_for_stoch_uniform.txt')
child_int = self.interactor.interactor_stochastic_uniform
child_int._view_items()
child_int.input_h.distribution.low = 30.0
child_int.input_h.distribution.high = 50.0
self.interactor.save_ranges_to_files({'stochastic_uniform': file_path})
# Read the saved file
file_object = open(file_path, 'r')
lines = file_object.readlines()
lines = [line.split() for line in lines]
file_object.close()
os.remove(file_path)
self.assertEqual(len(lines), 1)
self.assertEqual(len(lines[0]), 3)
self.assertEqual(lines[0][1], str(30.0))
self.assertEqual(lines[0][2], str(50.0))
def test_load_global_ranges_with_project_ranges(self):
""" Do global ranges get overwritten by the project settings ?
"""
code = "from blockcanvas.debug.my_operator import add,mul\n"\
"d = mul(a, b)\n" \
"e = mul(c, 3)\n" \
"f = add(d,e)"
self.block = Block(code)
self.context['c'] = 2.0
# This is to be done because the interactor doesn't get updated with
# change in block.
i_config = InteractorConfig(var_configs=[
VariableConfig(name='a', type='Shadow'),
VariableConfig(name='b', type='Parametric'),
VariableConfig(name='c', type='Shadow'),
VariableConfig(name='d', type='Parametric')
])
self.interactor = ConfigurableInteractor(block=self.block,
context=self.context,
interactor_config=i_config)
child_int = self.interactor.interactor_shadow
global_file_path = os.path.join(os.path.dirname(__file__), 'data',
'global_user_range.txt')
project_file_path = os.path.join(os.path.dirname(__file__), 'data',
'project_ranges.txt')
self.interactor.load_ranges_from_files({'shadow': project_file_path},
{'shadow': global_file_path})
child_int._view_items()
# Range of a is obtained from the project settings and not the
# global preferences
self.assertAlmostEqual(child_int.input_a__low, 3.4)
self.assertAlmostEqual(child_int.input_a__high, 5.0)
self.assertNotEqual(child_int.input_a__low, 0.15)
self.assertNotEqual(child_int.input_a__high, 9.3)
# Range of c is obtained from the global preferences
self.assertAlmostEqual(child_int.input_c__low, 2.0)
self.assertAlmostEqual(child_int.input_c__high, 7.0)
def test_load_global_ranges_without_project_ranges(self):
""" Do global ranges get used when there are no project settings ?
"""
code = "from blockcanvas.debug.my_operator import add,mul\n"\
"d = mul(a, b)\n" \
"e = mul(c, 3)\n" \
"f = add(d,e)"
self.block = Block(code)
self.context['c'] = 2.0
# This is to be done because the interactor doesn't get updated with
# change in block.
i_config = InteractorConfig(var_configs=[
VariableConfig(name='a', type='Shadow'),
VariableConfig(name='b', type='Parametric'),
VariableConfig(name='c', type='Shadow'),
VariableConfig(name='d', type='Parametric')
])
self.interactor = ConfigurableInteractor(block=self.block,
context=self.context,
interactor_config=i_config)
child_int = self.interactor.interactor_shadow
global_file_path = os.path.join(os.path.dirname(__file__), 'data',
'global_user_range.txt')
self.interactor.load_ranges_from_files({},
{'shadow': global_file_path})
child_int._view_items()
# Range of a is obtained from the project settings and not the
# global preferences
self.assertAlmostEqual(child_int.input_a__low, 0.15)
self.assertAlmostEqual(child_int.input_a__high, 9.3)
# Range of c is obtained from the global preferences
self.assertAlmostEqual(child_int.input_c__low, 2.0)
self.assertAlmostEqual(child_int.input_c__high, 7.0)
'from numpy import zeros\n'\
'from codetools.contexts.with_mask import Mask\n'\
'with Mask((depth < 4000.0) & (depth > 1000.0)):vp=1.5 ; vs=1.0'
## # Expanded form of with statement taken from PEP 343. This is just for testing
## code = 'from numpy import zeros\n'\
## 'array_len = depth.shape\n'\
## 'vp = zeros(array_len)\n'\
## 'vs = zeros(array_len)\n'\
## 'from codetools.contexts.with_mask import Mask\n'\
## 'mgr = Mask((depth < 4000.0) & (depth > 1000.0))\n'\
## 'exit_code = mgr.__exit__\n'\
## 'mgr.__enter__()\n'\
## 'exc = True\n'\
## 'try:\n'\
## ' try:\n'\
## ' vp = 1.0\n'\
## ' vs = 1.5\n'\
## ' except:\n'\
## ' exc = False\n'\
## 'finally:\n'\
## ' if exc:\n'\
## ' exit_code(None, None, None)'
b = Block(code)
b.execute(context)
print 'vp', context['vp']
print 'vs', context['vs']
### EOF ------------------------------------------------------------------------
def test_tracebacks(self):
'Tracebacks have correct file names and line numbers'
# If we want tracebacks to make sense, then the reported file names and
# line numbers need to associate with the code being executed
# regardless which block represents and executes the code.
def test(tb, lineno, filename):
self.assertEqual(tb.tb_lineno, lineno + 1)
self.assertEqual(tb.tb_frame.f_code.co_filename, filename)
def tracebacks():
"A list of the current exception's traceback objects."
tb = sys.exc_info()[2]
l = [tb]
while tb.tb_next is not None:
tb = tb.tb_next
l.append(tb)
return l
class File(StringIO, object):
"Extend StringIO with a 'name' attribute."
def __init__(self, name, *args, **kw):
super(File, self).__init__(*args, **kw)
self.name = name
a = Block(File('foo/a.py', 'y = x'))
try:
a.execute({})
except NameError as e:
test(tracebacks()[-1], 1, 'foo/a.py')
del a
a = Block(File('foo/a.py', 'import sys\ny = x'))
try:
a.execute({})
except NameError as e:
test(tracebacks()[-1], 2, 'foo/a.py')
#del a # (use below in 'Compose')
b = Block(
File('foo/b.py',
"import re\nl=re.findall('a(.*?)a', 'abacada')\nx = l[2]"))
try:
b.execute({})
except IndexError as e:
test(tracebacks()[-1], 3, 'foo/b.py')
#del b # (use below in 'Compose')
# Compose
c = Block((a, b))
try:
c.execute({})
except NameError as e:
test(tracebacks()[-1], 2, 'foo/a.py')
try:
c.execute({'x': 0})
except IndexError as e:
test(tracebacks()[-1], 3, 'foo/b.py')
del a, b, c
# Restrict
a = Block(File('foo/a.py', 'a = 0\nb = 1\nc = 2/a'))
try:
a.restrict(outputs='c').execute({})
except ZeroDivisionError as e:
test(tracebacks()[-1], 3, 'foo/a.py')
del a
# Throw out a sub-block
a = Block(File('foo/a.py', 'a = 0\nb = 1\nc = 2/a'))
a.sub_blocks.pop(1)
try:
a.execute({})
except ZeroDivisionError as e:
test(tracebacks()[-1], 3, 'foo/a.py')
del a
# Swap sub-blocks between blocks
a = Block(File('foo/a.py', 'a = 0\nb = 1'))
b = Block(File('foo/b.py', 'c = 2\nd = x'))
a.sub_blocks = b.sub_blocks
try:
a.execute({})
except NameError as e:
test(tracebacks()[-1], 2, 'foo/b.py')
a.sub_blocks = b.sub_blocks[:]
try:
a.execute({})
except NameError as e:
test(tracebacks()[-1], 2, 'foo/b.py')
del a, b
# from scimath.units.length import meter
code = "from blockcanvas.debug.my_operator import add, mul\n" \
"c = add(a,b)\n" \
"d = mul(c, 2)\n" \
"e = mul(c, 3)\n" \
"f = add(d,e)"
block=Block(code)
# context = ParametricContext(GeoContext(), {})
# context['a'] = UnitArray((1,2,3), units=meter)
# context['b'] = UnitArray((2,3,4), units=meter)
#
# interactor = ParametricInteractor(context=context, block = block)
# interactor.configure_traits()
# Context setup.
context = MultiContext(DataContext(name='Data'),{})
context['a'] = 1
context['b'] = 2
interactor = ParametricInteractor(context=context, block = block)
interactor.configure_traits()
for shadow in context.shadows:
block.execute(shadow)
print shadow['f'], 'Check with:', 5*(shadow['a']+shadow['b'])
### EOF ------------------------------------------------------------------------
# from scimath.units.length import meter
code = "from blockcanvas.debug.my_operator import add, mul\n" \
"c = add(a,b)\n" \
"d = mul(c, 2)\n" \
"e = mul(c, 3)\n" \
"f = add(d,e)"
block = Block(code)
# context = ParametricContext(GeoContext(), {})
# context['a'] = UnitArray((1,2,3), units=meter)
# context['b'] = UnitArray((2,3,4), units=meter)
#
# interactor = ParametricInteractor(context=context, block = block)
# interactor.configure_traits()
# Context setup.
context = MultiContext(DataContext(name='Data'), {})
context['a'] = 1
context['b'] = 2
interactor = ParametricInteractor(context=context, block=block)
interactor.configure_traits()
for shadow in context.shadows:
block.execute(shadow)
print shadow['f'], 'Check with:', 5 * (shadow['a'] + shadow['b'])
### EOF ------------------------------------------------------------------------
class ParametricInteractorTestCase(unittest.TestCase):
""" Unit testing for ParametricInteractor
"""
def setUp(self):
code = "from blockcanvas.debug.my_operator import add, mul\n" \
"c = add(a,b)\n" \
"d = mul(c, 2)\n" \
"e = mul(c, 3)\n" \
"f = add(d,e)"
self.block = Block(code)
# Context setup.
self.context = MultiContext(DataContext(name='Data'), {})
self.context['a'] = 1
self.context['b'] = 2
def test_attributes(self):
""" Test if creation of attributes is working correctly
"""
interactor = ParametricInteractor(context=self.context,
block=self.block)
self.assertTrue(hasattr(interactor, interactor._input_prefix + "a"))
self.assertTrue(hasattr(interactor, interactor._input_prefix + "b"))
# Check if the parameter items were written correctly
attribute_a = getattr(interactor, interactor._input_prefix + 'a')
attribute_b = getattr(interactor, interactor._input_prefix + 'b')
self.assertEqual(attribute_a.step, 0)
self.assertEqual(attribute_a.low, 1)
self.assertEqual(attribute_a.high, 1)
self.assertEqual(attribute_b.step, 0)
self.assertEqual(attribute_b.low, 2)
self.assertEqual(attribute_b.high, 2)
# Check if the inputs in the interactor are right
self.assertEqual(interactor.inputs, list(self.block.inputs))
def test_create_shadows(self):
""" Test if creating shadows is working correctly
"""
interactor = ParametricInteractor(context=self.context,
block=self.block)
# Change the parameters for the attributes
attribute_a = getattr(interactor, interactor._input_prefix + 'a')
attribute_b = getattr(interactor, interactor._input_prefix + 'b')
attribute_a.high = 4
attribute_a.step = 1
attribute_b.high = 3
attribute_b.step = 1
# Create the shadows.
interactor._update_contexts_button_changed()
# Check if the shadows were created correctly.
self.assertEqual(
len(self.context.shadows),
len(attribute_a.output_list) * len(attribute_b.output_list))
# Check if the shadow context gives desired result.
self.block.execute(self.context.shadows[-1])
self.assertEqual(
self.context.shadows[-1]['f'], 5 *
(self.context.shadows[-1]['a'] + self.context.shadows[-1]['b']))
a.execute({})
except NameError, e:
test(tracebacks()[-1], 1, 'foo/a.py')
del a
a = Block(File('foo/a.py', 'import sys\ny = x'))
try:
a.execute({})
except NameError, e:
test(tracebacks()[-1], 2, 'foo/a.py')
#del a # (use below in 'Compose')
b = Block(File('foo/b.py',
"import re\nl=re.findall('a(.*?)a', 'abacada')\nx = l[2]"))
try:
b.execute({})
except IndexError, e:
test(tracebacks()[-1], 3, 'foo/b.py')
#del b # (use below in 'Compose')
# Compose
c = Block((a,b))
try:
c.execute({})
except NameError, e:
test(tracebacks()[-1], 2, 'foo/a.py')
try:
c.execute({'x':0})
except IndexError, e:
test(tracebacks()[-1], 3, 'foo/b.py')
del a,b,c
