[incubator-nlpcraft] 01/02: Tests infrastructure. Initial commit.

[se...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

sergeykamov pushed a commit to branch NLPCRAFT-16
in repository https://gitbox.apache.org/repos/asf/incubator-nlpcraft.git

commit 7844b279927ced6a3b10c09758b516f935c40242
Author: Sergey Kamov <se...@apache.org>
AuthorDate: Sun Mar 15 13:05:46 2020 +0300

    Tests infrastructure. Initial commit.
---
 .../mgrs/nlp/enrichers/NCEnricherBaseSpec.scala    |  56 +++++++++++
 .../mgrs/nlp/enrichers/NCEnricherTestModel.scala   |  66 +++++++++++++
 .../mgrs/nlp/enrichers/NCEnrichersTestBeans.scala  | 108 +++++++++++++++++++++
 .../nlp/enrichers/sort/NCEnricherSortSpec.scala    |  38 ++++++++
 4 files changed, 268 insertions(+)

diff --git a/src/test/scala/org/apache/nlpcraft/probe/mgrs/nlp/enrichers/NCEnricherBaseSpec.scala b/src/test/scala/org/apache/nlpcraft/probe/mgrs/nlp/enrichers/NCEnricherBaseSpec.scala
new file mode 100644
index 0000000..07ef885
--- /dev/null
+++ b/src/test/scala/org/apache/nlpcraft/probe/mgrs/nlp/enrichers/NCEnricherBaseSpec.scala
@@ -0,0 +1,56 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.apache.nlpcraft.probe.mgrs.nlp.enrichers
+
+import org.apache.nlpcraft.model.tools.test.{NCTestClient, NCTestClientBuilder}
+import org.apache.nlpcraft.probe.embedded.NCEmbeddedProbe
+import org.junit.jupiter.api.Assertions.{assertEquals, assertTrue}
+import org.junit.jupiter.api.{AfterEach, BeforeEach}
+
+
+class NCEnricherBaseSpec {
+    private var client: NCTestClient = null
+
+    @BeforeEach
+    private[enrichers] def setUp(): Unit = {
+        NCEmbeddedProbe.start(classOf[NCEnricherTestModel])
+
+        client = new NCTestClientBuilder().newBuilder.build
+
+        client.open(NCEnricherTestModel.ID)
+    }
+
+    @AfterEach
+    private[enrichers] def tearDown(): Unit = {
+        client.close()
+    }
+
+    /**
+      *
+      * @param reqTxt
+      * @param expToks
+      */
+    private[enrichers] def check(reqTxt: String, expToks: NCTestToken*): Unit = {
+        val res = client.ask(reqTxt)
+
+        assertTrue(res.isOk)
+        assertTrue(res.getResult.isPresent)
+
+        assertEquals(NCTestSentence(expToks), NCTestSentence.deserialize(res.getResult.get()))
+    }
+}
diff --git a/src/test/scala/org/apache/nlpcraft/probe/mgrs/nlp/enrichers/NCEnricherTestModel.scala b/src/test/scala/org/apache/nlpcraft/probe/mgrs/nlp/enrichers/NCEnricherTestModel.scala
new file mode 100644
index 0000000..dbfc435
--- /dev/null
+++ b/src/test/scala/org/apache/nlpcraft/probe/mgrs/nlp/enrichers/NCEnricherTestModel.scala
@@ -0,0 +1,66 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.apache.nlpcraft.probe.mgrs.nlp.enrichers
+
+import java.util
+import java.util.Collections
+
+import org.apache.nlpcraft.model.{NCContext, NCElement, NCModelAdapter, NCResult}
+
+import scala.collection.JavaConverters._
+import scala.language.implicitConversions
+
+import NCEnricherTestModel._
+
+/**
+  * Enrichers test model.
+  */
+class NCEnricherTestModel extends NCModelAdapter(ID, "Model enrichers test", "1.0") {
+    private implicit def convert(s: String): NCResult = NCResult.text(s)
+
+    override def getElements: util.Set[NCElement] =
+        Set(
+            mkElement("A", Seq("A")),
+            mkElement("B", Seq("B")),
+            mkElement("C", Seq("C")),
+            mkElement("A B", Seq("A B")),
+            mkElement("B C", Seq("B C")),
+            mkElement("A B C", Seq("A B C")),
+            mkElement("D1", Seq("D")),
+            mkElement("D2", Seq("D"))
+        ).asJava
+
+    private def mkElement(id: String, syns: Seq[String]): NCElement =
+        new NCElement {
+            override def getId: String = id
+            override def getSynonyms: util.List[String] = syns.asJava
+            override def getGroups: util.List[String] = Collections.singletonList(GROUP)
+        }
+
+    // TODO: multiple variant.
+    override def onContext(ctx: NCContext): NCResult =
+        NCResult.text(
+            NCTestSentence.serialize(NCTestSentence(ctx.getVariants.asScala.head.asScala.map(t ⇒ NCTestToken(t))))
+        )
+}
+
+object NCEnricherTestModel {
+    final val ID = "test.enricher"
+
+    private final val GROUP = "test-enricher-group"
+}
diff --git a/src/test/scala/org/apache/nlpcraft/probe/mgrs/nlp/enrichers/NCEnrichersTestBeans.scala b/src/test/scala/org/apache/nlpcraft/probe/mgrs/nlp/enrichers/NCEnrichersTestBeans.scala
new file mode 100644
index 0000000..b8ee248
--- /dev/null
+++ b/src/test/scala/org/apache/nlpcraft/probe/mgrs/nlp/enrichers/NCEnrichersTestBeans.scala
@@ -0,0 +1,108 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.apache.nlpcraft.probe.mgrs.nlp.enrichers
+
+import java.util
+
+import com.google.gson.Gson
+import com.google.gson.reflect.TypeToken
+import org.apache.nlpcraft.model.NCToken
+import scala.collection.JavaConverters._
+import scala.language.implicitConversions
+
+sealed trait NCTestNote {
+    def id: String
+}
+
+case class NCTestSort(
+    asc: Option[Boolean],
+    subjNotes: java.util.List[String],
+    subjIndexes: java.util.List[Int],
+    byNotes: Option[java.util.List[String]],
+    byIndexes: Option[java.util.List[Int]]
+) extends NCTestNote {
+    override def id: String = "nlpcraft:sort"
+}
+
+case class NCTestRelation(`type`: String, indexes: java.util.List[Int], note: String) extends NCTestNote {
+    override def id: String = "nlpcraft:relation"
+}
+
+case class NCTestCommonNote(id: String) extends NCTestNote
+
+// TODO: add others
+case class NCTestToken(
+    origText: String,
+    isStop: Boolean,
+    note: NCTestNote
+)
+
+object NCTestToken {
+    def apply(t: NCToken): NCTestToken = {
+        val note: NCTestNote =
+            t.getId match {
+                case "nlpcraft:nlp" | "nlpcraft:coordinate" | "nlpcraft:num" | "nlpcraft:date" ⇒
+                    NCTestCommonNote(t.getId)
+
+                case "nlpcraft:sort" ⇒
+                    NCTestSort(
+                        asc = t.meta("asc"),
+                        subjNotes = t.meta("subjnotes"),
+                        subjIndexes = t.meta("subjindexes"),
+                        byNotes = t.meta("bynotes"),
+                        byIndexes = t.meta("byindexes")
+                    )
+                case "nlpcraft:relation" ⇒
+                    NCTestRelation(
+                        `type` = t.meta("type"),
+                        indexes = t.meta("indexes"),
+                        note = t.meta("note")
+                    )
+
+                case _ ⇒ throw new AssertionError(s"Unsupported token: ${t.getId}")
+            }
+
+        NCTestToken(t.getOriginalText, t.isStopWord, note)
+    }
+
+    def apply(origText: String, isStop: Boolean): NCTestToken =
+        new NCTestToken(origText, isStop, NCTestCommonNote("nlpcraft:nlp"))
+}
+
+case class NCTestSentence(tokens: Seq[NCTestToken])
+
+object NCTestSentence {
+    private val GSON = new Gson()
+    private val TYPE = new TypeToken[NCTestSentence]() {}.getType
+
+    def serialize(seq: NCTestSentence): String = {
+        GSON.toJson(seq)
+    }
+
+    def deserialize(s: String): NCTestSentence = {
+        GSON.fromJson(s, TYPE)
+    }
+}
+
+object a extends App {
+    val s = NCTestSentence.serialize(NCTestSentence(Seq(NCTestToken(origText = "text", isStop = false, note = NCTestRelation("a", Seq(1).asJava, "test")))))
+
+    println(s)
+
+    println(NCTestSentence.deserialize(s))
+}
diff --git a/src/test/scala/org/apache/nlpcraft/probe/mgrs/nlp/enrichers/sort/NCEnricherSortSpec.scala b/src/test/scala/org/apache/nlpcraft/probe/mgrs/nlp/enrichers/sort/NCEnricherSortSpec.scala
new file mode 100644
index 0000000..9d1ae95
--- /dev/null
+++ b/src/test/scala/org/apache/nlpcraft/probe/mgrs/nlp/enrichers/sort/NCEnricherSortSpec.scala
@@ -0,0 +1,38 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.apache.nlpcraft.probe.mgrs.nlp.enrichers.sort
+
+import org.apache.nlpcraft.probe.mgrs.nlp.enrichers.{NCEnricherBaseSpec, NCTestRelation, NCTestToken}
+import org.junit.jupiter.api.Test
+
+import scala.collection.JavaConverters._
+
+class NCEnricherSortSpec extends NCEnricherBaseSpec {
+
+    /**
+      *
+      * @throws Exception
+      */
+    @Test
+    private def test(): Unit = {
+        check(
+            "sort a1 by b1",
+            NCTestToken(origText = "text", isStop = false, note = NCTestRelation("a", Seq(1).asJava, "test"))
+        )
+    }
+}