Skip to content

Commit 75a638d

Browse files
committed
update modelmanager tests
1 parent fe11e07 commit 75a638d

2 files changed

Lines changed: 33 additions & 0 deletions

File tree

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,3 @@
1+
<manifest xmlns:android="http://schemas.android.com/apk/res/android">
2+
<uses-permission android:name="android.permission.INTERNET"/>
3+
</manifest>

ml/src/androidTest/kotlin/com/fpf/smartscansdk/ml/models/ModelManagerTest.kt

Lines changed: 30 additions & 0 deletions
Original file line numberDiff line numberDiff 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
}

0 commit comments

Comments
 (0)