adding tmpdir functionality to tests

metobscommon/tests/archive/unit_tests.py

@@ -3,21 +3,25 @@ import db
 from sqlalchemy import create_engine
 from sqlalchemy.orm import sessionmaker
 import mock
+import pytest
 
 import os
 
 #test database name
 DATABASE = 'test.db'
 
+
 class TestCreate(object):
     """Test create functionary."""
-
-    def setup(self):
-        if os.path.isfile(DATABASE):
-            os.remove(DATABASE)
+    @pytest.fixture(autouse=True)
+    def setup(self, tmpdir):
+        self.tmpdir = tmpdir.strpath
+        self.database = self.tmpdir = '/' + DATABASE
+        if os.path.isfile(self.database):
+            os.remove(self.database)
 
     def teardown(self):
-        os.remove(DATABASE)
+        os.remove(self.database)
 
     def setup_class(cls):
         pass
@@ -34,15 +38,15 @@ class TestCreate(object):
     def test_create_file_exists(self):
         """Test create functionality by checking if the database file exists."""
         #simulate 'python db.py test.db create'
-        args = db.parse_args([DATABASE, 'create'])
+        args = db.parse_args([self.database, 'create'])
         args.func(args)
         
-        assert os.path.exists(DATABASE)
+        assert os.path.exists(self.database)
         
     def test_create_query(self):
         """Test create functionality by querying the entire database and comparing results."""
         #simulate 'python db.py test.db create'
-        args = db.parse_args([DATABASE, 'create'])
+        args = db.parse_args([self.database, 'create'])
         args.func(args)
         
         #test input for querying Instrument, Site, Experiment, and FileType
@@ -74,7 +78,7 @@ class TestCreate(object):
         with mock.patch.object(db, 'input', lambda x: next(test_input_generator)):
         
             #simulate 'python db.py test.db query -i -s -e -t'
-            args = db.parse_args([DATABASE, 'query', '-i', '-s', '-e', '-t'])
+            args = db.parse_args([self.database, 'query', '-i', '-s', '-e', '-t'])
             queries = args.func(args)
         
         #find how many query results are found total