Skip to content

Commit e482d0f

Browse files
committed
Merge pull request #740 from pymc-devs/sqlite-tmpdir
test_sqlite_backend.py: Don't hardcode /tmp
2 parents b0b79f7 + 92a9457 commit e482d0f

File tree

1 file changed

+12
-8
lines changed

1 file changed

+12
-8
lines changed

pymc3/tests/test_sqlite_backend.py

Lines changed: 12 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1,56 +1,60 @@
11
import numpy.testing as npt
2+
import os
23
from pymc3.tests import backend_fixtures as bf
34
from pymc3.backends import ndarray, sqlite
5+
import tempfile
6+
7+
DBNAME = os.path.join(tempfile.gettempdir(), 'test.db')
48

59

610
class TestSQlite0dSampling(bf.SamplingTestCase):
711
backend = sqlite.SQLite
8-
name = '/tmp/test.db'
12+
name = DBNAME
913
shape = ()
1014

1115

1216
class TestSQlite1dSampling(bf.SamplingTestCase):
1317
backend = sqlite.SQLite
14-
name = '/tmp/test.db'
18+
name = DBNAME
1519
shape = 2
1620

1721

1822
class TestSQlite2dSampling(bf.SamplingTestCase):
1923
backend = sqlite.SQLite
20-
name = '/tmp/test.db'
24+
name = DBNAME
2125
shape = (2, 3)
2226

2327

2428
class TestSQLite0dSelection(bf.SelectionNoSliceTestCase):
2529
backend = sqlite.SQLite
26-
name = '/tmp/test.db'
30+
name = DBNAME
2731
shape = ()
2832

2933

3034
class TestSQLite1dSelection(bf.SelectionNoSliceTestCase):
3135
backend = sqlite.SQLite
32-
name = '/tmp/test.db'
36+
name = DBNAME
3337
shape = 2
3438

3539

3640
class TestSQLite2dSelection(bf.SelectionNoSliceTestCase):
3741
backend = sqlite.SQLite
38-
name = '/tmp/test.db'
42+
name = DBNAME
3943
shape = (2, 3)
4044

4145

4246
class TestSQLiteDumpLoad(bf.DumpLoadTestCase):
4347
backend = sqlite.SQLite
4448
load_func = staticmethod(sqlite.load)
45-
name = '/tmp/test.db'
49+
name = DBNAME
4650
shape = (2, 3)
4751

4852

4953
class TestNDArraySqliteEquality(bf.BackendEqualityTestCase):
5054
backend0 = ndarray.NDArray
5155
name0 = None
5256
backend1 = sqlite.SQLite
53-
name1 = '/tmp/test.db'
57+
name1 = DBNAME
5458
shape = (2, 3)
5559

5660

0 commit comments

Comments
 (0)