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_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_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(): 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(): 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_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_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"))