fix: replay recorded walks with selected identity backend
This commit is contained in:
parent
2798322417
commit
44badcf193
2 changed files with 196 additions and 13 deletions
|
|
@ -237,5 +237,41 @@ class CliTests(unittest.TestCase):
|
|||
load_index(root)
|
||||
|
||||
|
||||
class IdentityContractTests(unittest.TestCase):
|
||||
def test_missing_gallery_weights_exit_without_a_matcher_score(self):
|
||||
root = tempfile.mkdtemp()
|
||||
self.addCleanup(shutil.rmtree, root, ignore_errors=True)
|
||||
recording = os.path.join(root, "session")
|
||||
os.makedirs(recording)
|
||||
write_jpeg(os.path.join(recording, "000000.jpg"), np.zeros((8, 8, 3), dtype=np.uint8))
|
||||
with open(os.path.join(recording, "index.jsonl"), "w", encoding="utf-8") as handle:
|
||||
handle.write(json.dumps({"frame": 1, "path": "000000.jpg", "tracks": []}) + "\n")
|
||||
missing = os.path.join(root, "no-such-dinov2.safetensors")
|
||||
report_path = os.path.join(recording, "eval_live.json")
|
||||
env = os.environ.copy()
|
||||
env["DORAYAKI_DINOV2_WEIGHTS"] = missing
|
||||
completed = subprocess.run(
|
||||
[
|
||||
sys.executable,
|
||||
os.path.join(current_dir, "eval_live.py"),
|
||||
recording,
|
||||
"--replay",
|
||||
"--identity",
|
||||
"gallery",
|
||||
"--json",
|
||||
report_path,
|
||||
],
|
||||
capture_output=True,
|
||||
text=True,
|
||||
check=False,
|
||||
env=env,
|
||||
)
|
||||
self.assertEqual(completed.returncode, 2)
|
||||
payload = json.loads(completed.stdout)
|
||||
self.assertIn("DINOv2", payload["error"])
|
||||
self.assertIn("matcher", payload["error"])
|
||||
self.assertFalse(os.path.exists(report_path))
|
||||
self.assertNotIn("accepted_rate", completed.stdout)
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue