Skip to content

Commit

Permalink
feat: ability to import all *Gegevens classes
Browse files Browse the repository at this point in the history
This is mostly so that users can `from mdto.gegevensgroepen import *`,
and have all commonly used classes immediatly available.
  • Loading branch information
rien333 committed Dec 6, 2024
1 parent dd5dc7c commit 26c65e2
Show file tree
Hide file tree
Showing 2 changed files with 11 additions and 18 deletions.
15 changes: 8 additions & 7 deletions tests/test_mdto_voorbeelden.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
import pytest
import lxml.etree as ET
from mdto import Informatieobject, Bestand, from_file
import mdto
from mdto.gegevensgroepen import Informatieobject, Bestand


def serialization_chain(xmlfile: str) -> str:
Expand All @@ -15,7 +16,7 @@ def serialization_chain(xmlfile: str) -> str:
str: the re-serailized XML, as a string
"""
# Deserialize
object = from_file(xmlfile)
object = mdto.from_file(xmlfile)

# Serialize back to XML
output_tree = object.to_xml()
Expand All @@ -32,31 +33,31 @@ def serialization_chain(xmlfile: str) -> str:

def test_from_file_archiefstuk(voorbeeld_archiefstuk_xml):
"""Test that from_file() correctly parses Voorbeeld Archiefstuk Informatieobject.xml"""
archiefstuk = from_file(voorbeeld_archiefstuk_xml)
archiefstuk = mdto.from_file(voorbeeld_archiefstuk_xml)

assert isinstance(archiefstuk, Informatieobject)
assert archiefstuk.naam == "Verlenen kapvergunning Hooigracht 21 Den Haag"


def test_from_file_dossier(voorbeeld_dossier_xml):
"""Test that from_file() correctly parses Voorbeeld Dossier Informatieobject.xml"""
dossier = from_file(voorbeeld_dossier_xml)
dossier = mdto.from_file(voorbeeld_dossier_xml)

assert isinstance(dossier, Informatieobject)
assert dossier.trefwoord[1] == "kappen"


def test_from_file_serie(voorbeeld_serie_xml):
"""Test that from_file() correctly parses Voorbeeld Serie Informatieobject.xml"""
serie = from_file(voorbeeld_serie_xml)
serie = mdto.from_file(voorbeeld_serie_xml)

assert isinstance(serie, Informatieobject)
assert serie.naam == "Vergunningen van de gemeente 's-Gravenhage vanaf 1980"


def test_from_file_bestand(voorbeeld_bestand_xml):
"""Test that from_file() correctly parses Voorbeeld Bestand.xml"""
bestand = from_file(voorbeeld_bestand_xml)
bestand = mdto.from_file(voorbeeld_bestand_xml)

assert isinstance(bestand, Bestand)
assert (
Expand Down Expand Up @@ -99,7 +100,7 @@ def test_serialization_chain_bestand(voorbeeld_bestand_xml):

def test_file_saving(voorbeeld_archiefstuk_xml, tmp_path_factory):
"""Test if `save()` produces byte-for-byte equivalent XML from archiefstuk example"""
informatieobject = from_file(voorbeeld_archiefstuk_xml)
informatieobject = mdto.from_file(voorbeeld_archiefstuk_xml)

# location to write to
tmpdir = tmp_path_factory.mktemp("Output")
Expand Down
14 changes: 3 additions & 11 deletions tests/test_mdto_xsd.py
Original file line number Diff line number Diff line change
@@ -1,16 +1,8 @@
import pytest
import lxml.etree as ET
from pathlib import Path
from mdto import (
Informatieobject,
IdentificatieGegevens,
VerwijzingGegevens,
BeperkingGebruikGegevens,
DekkingInTijdGegevens,
GerelateerdInformatieobjectGegevens,
BegripGegevens,
create_bestand,
)
import mdto
from mdto.gegevensgroepen import *


def test_informatieobject_xml_validity(mdto_xsd):
Expand Down Expand Up @@ -69,7 +61,7 @@ def test_automatic_bestand_xml_validity(mdto_xsd, voorbeeld_archiefstuk_xml):
# use this .py file for automatic metadata generation
example_file = Path(__file__)
# create Bestand object from example_file + existing informatieobject
bestand = create_bestand(
bestand = mdto.create_bestand(
example_file,
"abcd-1234",
"Corsa",
Expand Down

0 comments on commit 26c65e2

Please sign in to comment.