forked from M-Labs/artiq
test_dataset_db: fix for windows
This commit is contained in:
parent
4f723e19a6
commit
37f14d94d0
|
@ -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()
|
||||
|
|
Loading…
Reference in New Issue