mirror of
https://github.com/TheAnachronism/docspell.git
synced 2025-06-22 18:38:26 +00:00
Convert unit tests to munit
This commit is contained in:
@ -1,6 +1,6 @@
|
||||
package docspell.analysis.classifier
|
||||
|
||||
import minitest._
|
||||
import munit._
|
||||
import cats.effect._
|
||||
import scala.concurrent.ExecutionContext
|
||||
import java.nio.file.Paths
|
||||
@ -10,7 +10,7 @@ import fs2.Stream
|
||||
import cats.data.Kleisli
|
||||
import TextClassifier.Data
|
||||
|
||||
object StanfordTextClassifierSuite extends SimpleTestSuite {
|
||||
class StanfordTextClassifierSuite extends FunSuite {
|
||||
val logger = Logger.log4s[IO](org.log4s.getLogger)
|
||||
|
||||
implicit val CS = IO.contextShift(ExecutionContext.global)
|
||||
|
@ -1,9 +1,9 @@
|
||||
package docspell.analysis.contact
|
||||
|
||||
import docspell.common.{NerLabel, NerTag}
|
||||
import minitest.SimpleTestSuite
|
||||
import munit._
|
||||
|
||||
object ContactAnnotateSpec extends SimpleTestSuite {
|
||||
class ContactAnnotateSpec extends FunSuite {
|
||||
|
||||
test("find email") {
|
||||
|
||||
|
@ -1,11 +1,11 @@
|
||||
package docspell.analysis.date
|
||||
|
||||
import docspell.files.TestFiles
|
||||
import minitest.SimpleTestSuite
|
||||
import munit._
|
||||
import docspell.common._
|
||||
import java.time._
|
||||
|
||||
object DateFindSpec extends SimpleTestSuite {
|
||||
class DateFindSpec extends FunSuite {
|
||||
|
||||
test("find simple dates") {
|
||||
val expect = Vector(
|
||||
|
@ -2,19 +2,17 @@ package docspell.analysis.nlp
|
||||
|
||||
import docspell.analysis.Env
|
||||
import docspell.common.Language.NLPLanguage
|
||||
import minitest.SimpleTestSuite
|
||||
import munit._
|
||||
import docspell.files.TestFiles
|
||||
import docspell.common._
|
||||
|
||||
object BaseCRFAnnotatorSuite extends SimpleTestSuite {
|
||||
class BaseCRFAnnotatorSuite extends FunSuite {
|
||||
|
||||
def annotate(language: NLPLanguage): String => Vector[NerLabel] =
|
||||
BasicCRFAnnotator.nerAnnotate(BasicCRFAnnotator.Cache.getAnnotator(language))
|
||||
|
||||
test("find english ner labels") {
|
||||
if (Env.isCI) {
|
||||
ignore("Test ignored on travis.")
|
||||
}
|
||||
assume(!Env.isCI, "Test ignored on travis.")
|
||||
|
||||
val labels = annotate(Language.English)(TestFiles.letterENText)
|
||||
val expect = Vector(
|
||||
@ -52,9 +50,7 @@ object BaseCRFAnnotatorSuite extends SimpleTestSuite {
|
||||
}
|
||||
|
||||
test("find german ner labels") {
|
||||
if (Env.isCI) {
|
||||
ignore("Test ignored on travis.")
|
||||
}
|
||||
assume(!Env.isCI, "Test ignored on travis.")
|
||||
|
||||
val labels = annotate(Language.German)(TestFiles.letterDEText)
|
||||
val expect = Vector(
|
||||
|
@ -4,22 +4,20 @@ import java.nio.file.Paths
|
||||
|
||||
import cats.effect.IO
|
||||
import docspell.analysis.Env
|
||||
import minitest.SimpleTestSuite
|
||||
import munit._
|
||||
import docspell.files.TestFiles
|
||||
import docspell.common._
|
||||
import docspell.common.syntax.FileSyntax._
|
||||
import edu.stanford.nlp.pipeline.StanfordCoreNLP
|
||||
|
||||
object StanfordNerAnnotatorSuite extends SimpleTestSuite {
|
||||
class StanfordNerAnnotatorSuite extends FunSuite {
|
||||
lazy val germanClassifier =
|
||||
new StanfordCoreNLP(Properties.nerGerman(None, false))
|
||||
lazy val englishClassifier =
|
||||
new StanfordCoreNLP(Properties.nerEnglish(None))
|
||||
|
||||
test("find english ner labels") {
|
||||
if (Env.isCI) {
|
||||
ignore("Test ignored on travis.")
|
||||
}
|
||||
assume(!Env.isCI, "Test ignored on travis.")
|
||||
|
||||
val labels =
|
||||
StanfordNerAnnotator.nerAnnotate(englishClassifier, TestFiles.letterENText)
|
||||
@ -58,9 +56,7 @@ object StanfordNerAnnotatorSuite extends SimpleTestSuite {
|
||||
}
|
||||
|
||||
test("find german ner labels") {
|
||||
if (Env.isCI) {
|
||||
ignore("Test ignored on travis.")
|
||||
}
|
||||
assume(!Env.isCI, "Test ignored on travis.")
|
||||
|
||||
val labels =
|
||||
StanfordNerAnnotator.nerAnnotate(germanClassifier, TestFiles.letterDEText)
|
||||
@ -83,9 +79,7 @@ object StanfordNerAnnotatorSuite extends SimpleTestSuite {
|
||||
}
|
||||
|
||||
test("regexner-only annotator") {
|
||||
if (Env.isCI) {
|
||||
ignore("Test ignored on travis.")
|
||||
}
|
||||
assume(!Env.isCI, "Test ignored on travis.")
|
||||
|
||||
val regexNerContent =
|
||||
s"""(?i)volantino ag${"\t"}ORGANIZATION${"\t"}LOCATION,PERSON,MISC${"\t"}3
|
||||
|
@ -1,8 +1,8 @@
|
||||
package docspell.analysis.split
|
||||
|
||||
import minitest.SimpleTestSuite
|
||||
import munit._
|
||||
|
||||
object TestSplitterSpec extends SimpleTestSuite {
|
||||
class TestSplitterSpec extends FunSuite {
|
||||
|
||||
test("simple splitting") {
|
||||
val text = """hiermit kündige ich meine Mitgliedschaft in der Kranken- und
|
||||
|
Reference in New Issue
Block a user