1 files changed, 8 insertions, 8 deletions
diff --git a/tests/nvme_id_ctrl_test.py b/tests/nvme_id_ctrl_test.py
index 0caf7b2..2810d94 100644
--- a/tests/nvme_id_ctrl_test.py
+++ b/tests/nvme_id_ctrl_test.py
@@ -1,4 +1,5 @@
-
+# SPDX-License-Identifier: GPL-2.0-or-later
+#
 # Copyright (c) 2015-2016 Western Digital Corporation or its affiliates.
 #
 # This program is free software; you can redistribute it and/or
@@ -27,7 +28,6 @@ NVMe Identify ctrl Testcase:-
 
 """
 
-from nose.tools import assert_equal
 from nvme_test import TestNVMe
 
 
@@ -37,20 +37,20 @@ class TestNVMeIdctrlCmd(TestNVMe):
     Represents Id ctrl testcase
     """
 
-    def __init__(self):
+    def setUp(self):
         """ Pre Section for TestNVMeIdctrlCmd. """
-        TestNVMe.__init__(self)
+        super().setUp()
         self.setup_log_dir(self.__class__.__name__)
 
-    def __del__(self):
+    def tearDown(self):
         """ Post Section for TestNVMeIdctrlCmd
 
             Call super class's destructor.
         """
-        TestNVMe.__del__(self)
+        super().tearDown()
 
     def test_id_ctrl(self):
         """ Testcase main """
         vendor = True
-        assert_equal(self.get_id_ctrl(), 0)
-        assert_equal(self.get_id_ctrl(vendor), 0)
+        self.assertEqual(self.get_id_ctrl(), 0)
+        self.assertEqual(self.get_id_ctrl(vendor), 0)