File tree Expand file tree Collapse file tree
kotlin/com/fpf/smartscansdk/ml/models Expand file tree Collapse file tree Original file line number Diff line number Diff line change 1+ <manifest xmlns : android =" http://schemas.android.com/apk/res/android" >
2+ <uses-permission android : name =" android.permission.INTERNET" />
3+ </manifest >
Original file line number Diff line number Diff line change @@ -89,4 +89,34 @@ class ModelManagerAndroidTest {
8989 assertTrue(listedModels.contains(modelName))
9090 }
9191 }
92+
93+ @Test
94+ fun testDownloadModelInternalAndLoadProvidersForAllModels () = runTest {
95+ ModelRegistry .forEach { (modelName, modelInfo) ->
96+
97+ val result = ModelManager .downloadModelInternal(context, modelInfo)
98+
99+ if (result.isFailure) {
100+ val exception = result.exceptionOrNull()
101+ fail(" Download failed for $modelName : ${exception?.message} " )
102+ }
103+
104+ assertTrue(ModelManager .modelExists(context, modelName))
105+
106+ when (modelInfo.type) {
107+ ModelType .TEXT_ENCODER -> {
108+ val provider = ModelManager .getTextEmbedder(context, modelName)
109+ assertNotNull(provider)
110+ }
111+ ModelType .IMAGE_ENCODER -> {
112+ val provider = ModelManager .getImageEmbedder(context, modelName)
113+ assertNotNull(provider)
114+ }
115+ ModelType .OBJECT_DETECTOR -> {
116+ val provider = ModelManager .getObjectDetector(context, modelName)
117+ assertNotNull(provider)
118+ }
119+ }
120+ }
121+ }
92122}
You can’t perform that action at this time.
0 commit comments