import pytest

from juritools.predict import JuriTagger, load_ner_model

from juritools.type import NamedEntity

from jurispacy_tokenizer import JuriSpacyTokenizer

import os

# Windows Fix for PosixPath issue

if os.name == "nt":

import pathlib

temp = pathlib.PosixPath

pathlib.PosixPath = pathlib.WindowsPath

FIXTURE_DIR = os.path.join(os.path.dirname(os.path.realpath(__file__)), "model")

model = load_ner_model(os.path.join(FIXTURE_DIR, "new_categories_model.pt"))

tokenizer = JuriSpacyTokenizer()

@pytest.fixture(scope="session")

def juritagger():

return JuriTagger(tokenizer, model)

def test_simple_text(juritagger):

]

@pytest.mark.skip(reason="output is not a list of dict anymore")

def test_simple_text_old(juritagger):

]