Add detect() unit test

6 years ago · 9aafe4aa24
5 changed files with 82 additions and 1 deletions
--- a/tests/.eslintrc
+++ b/tests/.eslintrc
@ -1,5 +1,6 @@
				@@ -1,5 +1,6 @@
 {
  "rules": {
-    "no-undef": 0
+    "no-undef": 0,
+    "camelcase": 0
  }
 }
--- a/tests/assets/traineddata/osd.traineddata
+++ b/tests/assets/traineddata/osd.traineddata
@ -0,0 +1,3 @@
				@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e19f2ae860792fdf372cf48d8ce70ae5da3c4052962fe22e9de1f680c374bb0e
+size 10562874
--- a/tests/assets/traineddata/osd.traineddata.gz
+++ b/tests/assets/traineddata/osd.traineddata.gz
@ -0,0 +1,3 @@
				@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:da43a9828b253a7b272c38635c20dbf6274ff7e2115488bcb54bc32acb8a9977
+size 4320198
--- a/tests/browser/detect.test.html
+++ b/tests/browser/detect.test.html
@ -0,0 +1,17 @@
				@@ -0,0 +1,17 @@
+<html>
+  <head>
+    <meta charset="utf-8">
+    <link rel="stylesheet" href="../../node_modules/mocha/mocha.css">
+  </head>
+  <body>
+    <div id="mocha"></div>
+    <script src="../../node_modules/mocha/mocha.js"></script>
+    <script src="../../node_modules/expect.js/index.js"></script>
+    <script src="/dist/tesseract.dev.js"></script>
+    <script>mocha.setup('bdd');</script>
+    <script src="../detect.test.js"></script>
+    <script>
+      mocha.run();
+    </script>
+  </body>
+</html>
--- a/tests/detect.test.js
+++ b/tests/detect.test.js
@ -0,0 +1,57 @@
				@@ -0,0 +1,57 @@
+const { TesseractWorker, utils: { loadLang } } = Tesseract;
+const IMAGE_PATH = 'http://localhost:3000/tests/assets/images';
+const loadLangOptions = {
+  langPath: 'http://localhost:3000/tests/assets/traineddata',
+  cachePath: './tests/assets/traineddata',
+};
+
+const getWorker = options => (
+  new TesseractWorker({
+    cacheMethod: 'readOnly',
+    ...loadLangOptions,
+    ...options,
+  })
+);
+
+before(function cb(done) {
+  this.timeout(30000);
+  const load = () => (
+    loadLang({
+      lang: 'osd',
+      cacheMethod: 'write',
+      ...loadLangOptions,
+    }).then(() => {
+      done();
+    })
+  );
+  if (typeof startServer !== 'undefined') {
+    startServer(load);
+  } else {
+    load();
+  }
+});
+
+after((done) => {
+  if (typeof stopServer !== 'undefined') {
+    stopServer(done);
+  } else {
+    done();
+  }
+});
+
+describe('detect()', () => {
+  it('should detect OSD', (done) => {
+    [
+      { name: 'cosmic.png', ans: { id: 12, degree: 0 } },
+    ].forEach(({ name, ans: { id, degree } }) => {
+      const worker = getWorker();
+      worker
+        .detect(`${IMAGE_PATH}/${name}`)
+        .then(({ tesseract_script_id, orientation_degrees }) => {
+          expect(tesseract_script_id).to.be(id);
+          expect(orientation_degrees).to.be(degree);
+          done();
+        });
+    });
+  }).timeout(10000);
+});