Skip to content

Commit ccdbf41

Browse files
committed
Clean up tests
1 parent 63a19e0 commit ccdbf41

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed
Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,21 @@
11
"""Evaluate examples of how to use the ndx-pose extension."""
22

33
import subprocess
4-
4+
from pathlib import Path
55

66
def test_example_usage_estimates_only():
77
"""Call examples/write_pose_estimates_only.py and check that it runs without errors."""
88
subprocess.run(["python", "examples/write_pose_estimates_only.py"], check=True)
99

10+
# Remove the generated test_pose.nwb if it exists
11+
if Path("test_pose.nwb").exists():
12+
Path("test_pose.nwb").unlink()
13+
1014

1115
def test_example_usage_training_only():
1216
"""Call examples/write_pose_training.py and check that it runs without errors."""
1317
subprocess.run(["python", "examples/write_pose_training.py"], check=True)
18+
19+
# Remove the generated test_pose.nwb if it exists
20+
if Path("test_pose.nwb").exists():
21+
Path("test_pose.nwb").unlink()

0 commit comments

Comments
 (0)