diff --git a/artiq/test/test_dataset_db.py b/artiq/test/test_dataset_db.py index 3d087a806..cafcf97d1 100644 --- a/artiq/test/test_dataset_db.py +++ b/artiq/test/test_dataset_db.py @@ -1,6 +1,7 @@ """Test internal dataset representation (persistence, applets)""" import unittest import tempfile +import os from artiq.master.databases import DatasetDB from sipyco import pyon @@ -11,11 +12,14 @@ KEY3 = "key3" DATA = list(range(10)) COMP = "gzip" +# tempfile.NamedTemporaryFile: +# use delete=False and manual cleanup +# for Windows compatibility class TestDatasetDB(unittest.TestCase): def setUp(self): # empty dataset persistance file - self.persist_file = tempfile.NamedTemporaryFile(mode="w+") + self.persist_file = tempfile.NamedTemporaryFile(mode="w+", delete=False) print("{}", file=self.persist_file, flush=True) self.ddb = DatasetDB(self.persist_file.name) @@ -26,6 +30,9 @@ class TestDatasetDB(unittest.TestCase): self.save_ddb_to_disk() + def tearDown(self): + os.unlink(self.persist_file.name) + def save_ddb_to_disk(self): self.ddb.save() self.persist_file.flush()