def test_add_field_with_clobber(): """Test adding a field with clobber on.""" fields = ModelDataFields() fields.new_field_location('node', 12) assert fields.add_empty('a', at='node') is not fields.add_empty('a', at='node', noclobber=False) assert fields.add_ones('b', at='node') is not fields.add_ones('b', at='node', noclobber=False) assert fields.add_zeros('c', at='node') is not fields.add_zeros('c', at='node', noclobber=False)
def test_add_field_with_clobber(): """Test adding a field with clobber on.""" fields = ModelDataFields() fields.new_field_location('node', 12) assert_is_not(fields.add_empty('a', at='node'), fields.add_empty('a', at='node', noclobber=False)) assert_is_not(fields.add_ones('b', at='node'), fields.add_ones('b', at='node', noclobber=False)) assert_is_not(fields.add_zeros('c', at='node'), fields.add_zeros('c', at='node', noclobber=False))
def test_grid_field_add_zeros_ones_empty(): """Test creating scalar fields with add_zeros, add_empty, and add_ones.""" fields = ModelDataFields() fields.new_field_location("grid", 1) with pytest.raises(ValueError): fields.add_zeros("value", at="grid") with pytest.raises(ValueError): fields.add_empty("value", at="grid") with pytest.raises(ValueError): fields.add_ones("value", at="grid")
def test_add_existing_field_with_noclobber(): """Test noclobber raises an error with an existing field.""" fields = ModelDataFields() fields.new_field_location('node', 12) fields.add_empty('z', at='node') with pytest.raises(FieldError): fields.add_empty('z', at='node', noclobber=True) with pytest.raises(FieldError): fields.add_ones('z', at='node', noclobber=True) with pytest.raises(FieldError): fields.add_zeros('z', at='node', noclobber=True)
def test_add_existing_field_with_noclobber(): """Test noclobber raises an error with an existing field.""" fields = ModelDataFields() fields.new_field_location("node", 12) fields.add_empty("z", at="node") with pytest.raises(FieldError): fields.add_empty("z", at="node", noclobber=True) with pytest.raises(FieldError): fields.add_ones("z", at="node", noclobber=True) with pytest.raises(FieldError): fields.add_zeros("z", at="node", noclobber=True)
def test_add_existing_field_with_clobber_off(): """Test clobber raises an error with an existing field.""" fields = ModelDataFields() fields.new_field_location("node", 12) fields.add_empty("z", at="node") with pytest.raises(FieldError): fields.add_empty("z", at="node", clobber=False) with pytest.raises(FieldError): fields.add_ones("z", at="node", clobber=False) with pytest.raises(FieldError): fields.add_zeros("z", at="node", clobber=False)
def test_add_existing_field_default(): """Test default is to not replace existing field.""" fields = ModelDataFields() fields.new_field_location("node", 12) fields.add_empty("z", at="node") with pytest.raises(FieldError): fields.add_empty("z", at="node") with pytest.raises(FieldError): fields.add_ones("z", at="node") with pytest.raises(FieldError): fields.add_zeros("z", at="node")
def test_add_field_with_clobber_off(): """Test clobber does not raise an error with an new field.""" fields = ModelDataFields() fields.new_field_location("node", 12) fields.add_empty("a", at="node", clobber=False) assert "a" in fields["node"] fields.add_ones("b", at="node", clobber=False) assert "b" in fields["node"] fields.add_zeros("c", at="node", clobber=False) assert "c" in fields["node"]
def test_add_field_with_noclobber(): """Test noclobber does not raise an error with an new field.""" fields = ModelDataFields() fields.new_field_location('node', 12) fields.add_empty('a', at='node', noclobber=True) assert_true('a' in fields['node']) fields.add_ones('b', at='node', noclobber=True) assert_true('b' in fields['node']) fields.add_zeros('c', at='node', noclobber=True) assert_true('c' in fields['node'])
def test_add_field_with_noclobber(): """Test noclobber does not raise an error with an new field.""" fields = ModelDataFields() fields.new_field_location('node', 12) fields.add_empty('a', at='node', noclobber=True) assert 'a' in fields['node'] fields.add_ones('b', at='node', noclobber=True) assert 'b' in fields['node'] fields.add_zeros('c', at='node', noclobber=True) assert 'c' in fields['node']
def test_add_field_with_noclobber(): """Test noclobber does not raise an error with an new field.""" fields = ModelDataFields() fields.new_field_location("node", 12) fields.add_empty("a", at="node", noclobber=True) assert "a" in fields["node"] fields.add_ones("b", at="node", noclobber=True) assert "b" in fields["node"] fields.add_zeros("c", at="node", noclobber=True) assert "c" in fields["node"]
def test_delete_field(): fields = ModelDataFields() fields.new_field_location('link', 17) assert_dict_equal(dict(), fields.at_link) assert_raises(AttributeError, lambda: fields.at_node) fields.add_zeros('link', 'vals') assert_array_equal(np.zeros(17), fields.at_link['vals']) assert_raises(KeyError, lambda: fields.delete_field('node', 'vals')) fields.delete_field('link', 'vals') assert_raises(KeyError, lambda: fields.field_units('link', 'vals')) assert_raises(KeyError, lambda: fields.at_link['vals'])
def test_add_field_with_clobber(): """Test adding a field with clobber on.""" fields = ModelDataFields() fields.new_field_location("node", 12) assert fields.add_empty("a", at="node") is not fields.add_empty( "a", at="node", noclobber=False ) assert fields.add_ones("b", at="node") is not fields.add_ones( "b", at="node", noclobber=False ) assert fields.add_zeros("c", at="node") is not fields.add_zeros( "c", at="node", noclobber=False )
def test_add_field_with_clobber(): """Test adding a field with clobber on.""" fields = ModelDataFields() fields.new_field_location("node", 12) assert fields.add_empty("a", at="node") is not fields.add_empty("a", at="node", clobber=True) assert fields.add_ones("b", at="node") is not fields.add_ones("b", at="node", clobber=True) assert fields.add_zeros("c", at="node") is not fields.add_zeros("c", at="node", clobber=True)
def test_delete_field(): fields = ModelDataFields() fields.new_field_location('link', 17) assert dict() == fields.at_link with pytest.raises(AttributeError): fields.at_node fields.add_zeros('link', 'vals') assert_array_equal(np.zeros(17), fields.at_link['vals']) with pytest.raises(KeyError): fields.delete_field('node', 'vals') fields.delete_field('link', 'vals') with pytest.raises(KeyError): fields.field_units('link', 'vals') with pytest.raises(KeyError): fields.at_link['vals']
def test_delete_field(): fields = ModelDataFields() fields.new_field_location("link", 17) assert dict() == fields.at_link with pytest.raises(AttributeError): fields.at_node fields.add_zeros("vals", at="link") assert_array_equal(np.zeros(17), fields.at_link["vals"]) with pytest.raises(KeyError): fields.delete_field("node", "vals") fields.delete_field("link", "vals") with pytest.raises(KeyError): fields.field_units("link", "vals") with pytest.raises(KeyError): fields.at_link["vals"]
def test_delete_field(): fields = ModelDataFields() fields.new_field_location("link", 17) assert dict() == fields.at_link with pytest.raises(AttributeError): fields.at_node fields.add_zeros("link", "vals") assert_array_equal(np.zeros(17), fields.at_link["vals"]) with pytest.raises(KeyError): fields.delete_field("node", "vals") fields.delete_field("link", "vals") with pytest.raises(KeyError): fields.field_units("link", "vals") with pytest.raises(KeyError): fields.at_link["vals"]