Add SqueezeNet test

Author: EmergentOrder

.travis.yml

@@ -1,5 +1,5 @@
 language: scala
 script: 
   - bash get_models.sh
-  - travis_wait 30 sbt +test -J-Xmx5G
+  - travis_wait 30 sbt test -J-Xmx5G
 #  - sbt docsJVM/mdoc

backends/.jvm/src/test/scala/SqueezeNetTest.scala

@@ -0,0 +1,42 @@
+package org.emergentorder.onnx.backends
+
+import java.nio.file.{Files, Paths}
+import org.emergentorder.onnx.Tensors._
+import org.emergentorder.onnx.backends._
+import org.emergentorder.compiletime._
+import io.kjaer.compiletime._
+
+
+import org.scalatest.flatspec.AnyFlatSpec
+import org.scalatest.matchers.should._
+
+
+class ONNXScalaSpec extends AnyFlatSpec with Matchers{
+
+  "SqueezeNet ONNX-Scala model" should "predict dummy image class" in {
+    val squeezenetBytes = Files.readAllBytes(Paths.get("squeezenet1.1.onnx"))
+    val squeezenet = new ORTModelBackend(squeezenetBytes)
+    val data = Array.fill(1*3*224*224){42f}
+    //In NCHW tensor image format
+    val shape =                    1     #:     3      #:    224    #: 224     #: SNil
+    val tensorShapeDenotation = "Batch" ##: "Channel" ##: "Height" ##: "Width" ##: TSNil
+
+    val tensorDenotation: String & Singleton = "Image"
+
+    val imageTens = Tensor(data,tensorDenotation,tensorShapeDenotation,shape)
+
+    //or as a shorthand if you aren't concerned with enforcing denotations
+    val imageTensDefaultDenotations = Tensor(data,shape)
+    val out = squeezenet.fullModel[Float, 
+                               "ImageNetClassification",
+                               "Batch" ##: "Class" ##: TSNil,
+                               1 #: 1000 #: SNil](Tuple(imageTens))
+
+    //The output shape
+    assert(out.shape(0) == 1)
+    assert(out.shape(1) == 1000)
+
+    //The highest probability (predicted) class
+    assert(out.data.indices.maxBy(out.data) == 418)
+  }
+}

build.sbt

@@ -3,6 +3,7 @@ import sbtcrossproject.CrossPlugin.autoImport.{crossProject, CrossType}
 val dottyVersion = "3.0.0-RC1"
 val scala213Version = "2.13.5"
 val spireVersion = "0.17.0"
+val scalaTestVersion = "3.2.5"
 
 scalaVersion := dottyVersion
 
@@ -50,11 +51,11 @@ lazy val proto = (crossProject(JSPlatform, JVMPlatform)
     PB.protoSources in Compile := Seq(file("proto/src/main/protobuf")),
   )
 
-lazy val backends = (crossProject(JSPlatform, JVMPlatform)
+lazy val backends = (crossProject(JVMPlatform, JSPlatform)
   .crossType(CrossType.Pure) in file("backends"))
   .dependsOn(core)
 //conditionally enabling/disable based on version, still not working
-  .enablePlugins(ScalaJSBundlerPlugin)//, ScalablyTypedConverterPlugin)
+//  .enablePlugins(ScalaJSBundlerPlugin)//, ScalablyTypedConverterPlugin)
   .settings(
     commonSettings,
     name := "onnx-scala-backends",
@@ -77,10 +78,12 @@ lazy val backends = (crossProject(JSPlatform, JVMPlatform)
     ),
     crossScalaVersions := Seq(dottyVersion, scala213Version)
   )
-.jvmSettings().jsSettings(
-      scalaJSUseMainModuleInitializer := true, //, //Testing
+.jvmSettings(
+libraryDependencies += ("org.scalatest" %% "scalatest" % scalaTestVersion) % Test,
+).jsSettings(
+      scalaJSUseMainModuleInitializer := true) //, //Testing
 //Seems to be a bundling issue, copying things manually seems to work
-     npmDependencies in Compile += "onnxjs" -> "0.1.8")
+//     npmDependencies in Compile += "onnxjs" -> "0.1.8")
 
 lazy val core = (crossProject(JSPlatform, JVMPlatform)
   .crossType(CrossType.Pure) in file("core"))