[verified] refresh OpenVINO router classifier prototype
This commit is contained in:
@@ -88,6 +88,14 @@ class RouterClassifierTests(unittest.TestCase):
|
||||
self.assertEqual(len(result["results"]), 2)
|
||||
self.assertGreater(result["npu_busy_delta_us"], 0)
|
||||
|
||||
def test_batch_limits_are_enforced(self):
|
||||
svc = self.service()
|
||||
with self.assertRaisesRegex(ValueError, "at least one"):
|
||||
svc.batch_classify([])
|
||||
too_many = [{"id": str(i), "text": "What time is it?"} for i in range(router_classifier.DEFAULT_MAX_BATCH_SIZE + 1)]
|
||||
with self.assertRaisesRegex(ValueError, "max_batch_size"):
|
||||
svc.batch_classify(too_many)
|
||||
|
||||
def test_fixture_file_is_valid_jsonl(self):
|
||||
fixture = ROOT / "fixtures" / "atlas_hermes_messages.jsonl"
|
||||
rows = [json.loads(line) for line in fixture.read_text().splitlines() if line.strip()]
|
||||
@@ -97,6 +105,17 @@ class RouterClassifierTests(unittest.TestCase):
|
||||
self.assertIn("text", row)
|
||||
self.assertIn("expected", row)
|
||||
|
||||
def test_synthetic_fixture_expectations(self):
|
||||
svc = self.service()
|
||||
fixture = ROOT / "fixtures" / "atlas_hermes_messages.jsonl"
|
||||
rows = [json.loads(line) for line in fixture.read_text().splitlines() if line.strip()]
|
||||
for row in rows:
|
||||
with self.subTest(row=row["id"]):
|
||||
result = svc.classify(row["id"], row["text"], {"include_evidence": False})
|
||||
labels = result["labels"]
|
||||
for label_name, expected_value in row["expected"].items():
|
||||
self.assertEqual(labels[label_name]["value"], expected_value)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
|
||||
Reference in New Issue
Block a user