Implement BO PreisblattNetznutzung (#325)

Author: hf-kklein

src/bo4e/bo/preisblattnetznutzung.py

@@ -0,0 +1,42 @@
+"""
+Contains PreisblattNetnutzung class and corresponding marshmallow schema for de-/serialization
+"""
+
+import attr
+from marshmallow_enum import EnumField  # type:ignore[import]
+
+from bo4e.bo.preisblatt import Preisblatt, PreisblattSchema
+from bo4e.enum.bilanzierungsmethode import Bilanzierungsmethode
+from bo4e.enum.botyp import BoTyp
+from bo4e.enum.kundengruppe import Kundengruppe
+from bo4e.enum.netzebene import Netzebene
+
+
+# pylint: disable=too-few-public-methods
+@attr.s(auto_attribs=True, kw_only=True)
+class PreisblattNetznutzung(Preisblatt):
+    """
+    Die Variante des Preisblattmodells zur Abbildung der Netznutzungspreise
+    """
+
+    bo_typ: BoTyp = attr.ib(default=BoTyp.PREISBLATTNETZNUTZUNG)
+    # required attributes (additional to those of Preisblatt)
+    #: Die Preise gelten für Marktlokationen der angebebenen Bilanzierungsmethode
+    bilanzierungsmethode: Bilanzierungsmethode = attr.ib(validator=attr.validators.instance_of(Bilanzierungsmethode))
+    #: Die Preise gelten für Marktlokationen in der angebebenen Netzebene
+    netzebene: Netzebene = attr.ib(validator=attr.validators.instance_of(Netzebene))
+    kundengruppe: Kundengruppe = attr.ib(validator=attr.validators.instance_of(Kundengruppe))
+
+    # there are no optional attributes (additionally to those of Preisblatt)
+
+
+class PreisblattNetznutzungSchema(PreisblattSchema):
+    """
+    Schema for de-/serialization of PreisblattNetznutzung
+    """
+
+    class_name = PreisblattNetznutzung  # type:ignore[assignment]
+    # required attributes
+    bilanzierungsmethode = EnumField(Bilanzierungsmethode)
+    netzebene = EnumField(Netzebene)
+    kundengruppe = EnumField(Kundengruppe)

tests/test_preisblattnetznutzung.py

@@ -0,0 +1,43 @@
+import pytest  # type:ignore[import]
+
+from bo4e.bo.preisblattnetznutzung import PreisblattNetznutzung, PreisblattNetznutzungSchema
+from bo4e.enum.bilanzierungsmethode import Bilanzierungsmethode
+from bo4e.enum.kundengruppe import Kundengruppe
+from bo4e.enum.netzebene import Netzebene
+from bo4e.enum.preisstatus import Preisstatus
+from bo4e.enum.sparte import Sparte
+from tests.serialization_helper import assert_serialization_roundtrip  # type:ignore[import]
+from tests.test_marktteilnehmer import example_marktteilnehmer  # type:ignore[import]
+from tests.test_preisposition import example_preisposition  # type:ignore[import]
+from tests.test_zeitraum import example_zeitraum  # type:ignore[import]
+
+
+class TestPreisblatt:
+    @pytest.mark.parametrize(
+        "preisblatt_netznutzung",
+        [
+            pytest.param(
+                PreisblattNetznutzung(
+                    bezeichnung="foo",
+                    sparte=Sparte.STROM,
+                    preisstatus=Preisstatus.ENDGUELTIG,
+                    preispositionen=[example_preisposition],
+                    gueltigkeit=example_zeitraum,
+                    herausgeber=example_marktteilnehmer,
+                    bilanzierungsmethode=Bilanzierungsmethode.TLP_GEMEINSAM,
+                    netzebene=Netzebene.MSP,
+                    kundengruppe=Kundengruppe.SLP_G_GKO,
+                )
+            ),
+        ],
+    )
+    def test_serialization_roundtrip(self, preisblatt_netznutzung: PreisblattNetznutzung):
+        """
+        Test de-/serialisation
+        """
+        assert_serialization_roundtrip(preisblatt_netznutzung, PreisblattNetznutzungSchema())
+
+    def test_missing_required_attribute(self):
+        with pytest.raises(TypeError) as excinfo:
+            _ = PreisblattNetznutzung()
+        assert "missing 8 required" in str(excinfo.value)  # 5 from preisblatt + 3 from preisblatt netznutzung