Merge nucleic/sleek-ember-seal-uady into dev
This commit is contained in:
@@ -1,4 +1,5 @@
|
||||
import argparse
|
||||
import json
|
||||
import sys
|
||||
import tempfile
|
||||
import unittest
|
||||
@@ -94,6 +95,36 @@ class CoreMLQuantizationConfigTests(unittest.TestCase):
|
||||
self.assertEqual([1.0], output["output"].tolist())
|
||||
self.assertEqual([], list(root.glob("*.mlpackage")))
|
||||
|
||||
result_package = Path(tempfile.mkdtemp(suffix=".mlpackage"))
|
||||
self.assertTrue(result_package.exists())
|
||||
|
||||
def test_activation_cache_requires_exact_contract(self):
|
||||
expected = {
|
||||
"schemaVersion": 1,
|
||||
"sourcePackageSha256": "abc",
|
||||
}
|
||||
with tempfile.TemporaryDirectory() as temp:
|
||||
root = Path(temp)
|
||||
package = root / "a8.mlpackage"
|
||||
package.mkdir()
|
||||
manifest = root / "a8-manifest.json"
|
||||
manifest.write_text(json.dumps(expected), encoding="utf-8")
|
||||
self.assertTrue(
|
||||
quantize_coreml._activation_cache_is_valid(
|
||||
package,
|
||||
manifest,
|
||||
expected,
|
||||
)
|
||||
)
|
||||
manifest.write_text("{}", encoding="utf-8")
|
||||
self.assertFalse(
|
||||
quantize_coreml._activation_cache_is_valid(
|
||||
package,
|
||||
manifest,
|
||||
expected,
|
||||
)
|
||||
)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
|
||||
Reference in New Issue
Block a user