unlink tmp files

aabouzaid · Sep 17, 2017 · 8465ed0 · 8465ed0
1 parent 0dc61e7
commit 8465ed0
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/netbox/tests/test_netbox.py b/netbox/tests/test_netbox.py
@@ -316,10 +316,6 @@ def test_open_yaml_file_exists(self, yaml_file):
         assert config_output["netbox"]
         assert config_output["netbox"]["main"]["api_url"]
 
-    @staticmethod
-    def teardown_function():
-        os.unlink(netbox_config_file.name)
-
     @pytest.mark.parametrize("yaml_file", [
         "nonexists.yml"
     ])
@@ -342,8 +338,9 @@ def test_open_yaml_file_invalid(self, yaml_file):
             netbox.open_yaml_file(yaml_file)
         assert invalid_yaml_syntax
 
-    @staticmethod
-    def teardown_function():
+    @classmethod
+    def teardown_class(cls):
+        os.unlink(netbox_config_file.name)
         os.unlink(netbox_config_file_invalid.name)