def test_add_multiple_groups():
fields = ModelDataFields()
fields.new_field_location('node', 12)
fields.new_field_location('cell', 2)
fields.new_field_location('face', 7)
fields.new_field_location('link', 7)
assert_set_equal(set(['node', 'cell', 'face', 'link']), fields.groups)
def test_getitem():
fields = ModelDataFields()
fields.new_field_location('node', 12)
assert_dict_equal(dict(), fields['node'])
assert_raises(GroupError, lambda k: fields[k], 'cell')
assert_raises(KeyError, lambda k: fields[k], 'cell')
def test_at_attribute():
fields = ModelDataFields()
fields.new_field_location('node', 12)
assert_dict_equal(dict(), fields.at_node)
assert_raises(AttributeError, lambda: fields.at_cell)
fields.add_ones('node', 'z')
assert_array_equal(np.ones(12), fields.at_node['z'])
def test_getitem(self):
fields = ModelDataFields()
fields.new_field_location('node', 12)
self.assertDictEqual(dict(), fields['node'])
with self.assertRaises(GroupError):
fields['cell']
with self.assertRaises(KeyError):
fields['cell']
def test_ones():
fields = ModelDataFields()
fields.new_field_location('node', 12)
fields.new_field_location('cell', 2)
value_array = fields.ones('node')
assert_array_equal(np.ones(12), value_array)
value_array = fields.ones('cell')
assert_array_equal(np.ones(2), value_array)
def test_at_attribute(self):
fields = ModelDataFields()
fields.new_field_location('node', 12)
self.assertDictEqual(dict(), fields.at_node)
with self.assertRaises(AttributeError):
fields.at_cell
fields.add_ones('node', 'z')
assert_array_equal(np.ones(12), fields.at_node['z'])
def test_add_ones():
fields = ModelDataFields()
fields.new_field_location('node', 12)
fields.new_field_location('cell', 2)
fields.add_ones('node', 'z')
assert_array_equal(np.ones(12), fields['node']['z'])
assert_array_equal(np.ones(12), fields.field_values('node', 'z'))
fields.add_ones('cell', 'z')
assert_array_equal(np.ones(2), fields['cell']['z'])
assert_array_equal(np.ones(2), fields.field_values('cell', 'z'))
def test_add_ones_return_value():
fields = ModelDataFields()
fields.new_field_location('node', 12)
fields.new_field_location('cell', 2)
rtn_value = fields.add_ones('node', 'z')
assert_array_equal(rtn_value, np.ones(12))
assert_is(rtn_value, fields['node']['z'])
assert_is(rtn_value, fields.field_values('node', 'z'))
rtn_value = fields.add_ones('cell', 'z')
assert_array_equal(rtn_value, np.ones(2))
assert_is(rtn_value, fields['cell']['z'])
assert_is(rtn_value, fields.field_values('cell', 'z'))
def test_add_existing_group():
fields = ModelDataFields()
fields.new_field_location('node', 12)
assert_raises(ValueError, fields.new_field_location, 'node', 24)
def test_new_field_location():
fields = ModelDataFields()
fields.new_field_location('node', 12)
assert_set_equal(set(['node']), fields.groups)
def test_init():
fields = ModelDataFields()
assert_set_equal(set(), fields.groups)
def test_has_group():
fields = ModelDataFields()
fields.new_field_location('node', 12)
assert_true(fields.has_group('node'))
assert_false(fields.has_group('cell'))
def test_has_group(self):
fields = ModelDataFields()
fields.new_field_location('node', 12)
self.assertTrue(fields.has_group('node'))
self.assertFalse(fields.has_group('cell'))
def test_add_existing_group(self):
fields = ModelDataFields()
fields.new_field_location('node', 12)
with self.assertRaises(ValueError):
fields.new_field_location('node', 24)
def test_new_field_location(self):
fields = ModelDataFields()
fields.new_field_location('node', 12)
self.assertSetEqual(set(['node']), fields.groups)
def test_init(self):
fields = ModelDataFields()
self.assertSetEqual(set(), fields.groups)