diff options
Diffstat (limited to 'tests/nvme_writezeros_test.py')
| -rw-r--r-- | tests/nvme_writezeros_test.py | 23 |
1 files changed, 12 insertions, 11 deletions
diff --git a/tests/nvme_writezeros_test.py b/tests/nvme_writezeros_test.py index ba3538c..3231e3d 100644 --- a/tests/nvme_writezeros_test.py +++ b/tests/nvme_writezeros_test.py @@ -1,3 +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 @@ -29,7 +31,6 @@ NVMe Write Zeros:- import filecmp -from nose.tools import assert_equal from nvme_test_io import TestNVMeIO @@ -46,9 +47,9 @@ class TestNVMeWriteZeros(TestNVMeIO): - test_log_dir : directory for logs, temp files. """ - def __init__(self): + def setUp(self): """ Pre Section for TestNVMeWriteZeros """ - TestNVMeIO.__init__(self) + super().setUp() self.start_block = 1023 self.block_count = 0 self.setup_log_dir(self.__class__.__name__) @@ -59,9 +60,9 @@ class TestNVMeWriteZeros(TestNVMeIO): self.create_data_file(self.zero_file, self.data_size, '\0') open(self.read_file, 'a').close() - def __del__(self): + def tearDown(self): """ Post Section for TestNVMeWriteZeros """ - TestNVMeIO.__del__(self) + super().tearDown() def write_zeroes(self): """ Wrapper for nvme write-zeroe @@ -96,9 +97,9 @@ class TestNVMeWriteZeros(TestNVMeIO): def test_write_zeros(self): """ Testcae main """ - assert_equal(self.nvme_write(), 0) - assert_equal(self.nvme_read(), 0) - assert_equal(self.validate_write_read(), 0) - assert_equal(self.write_zeroes(), 0) - assert_equal(self.nvme_read(), 0) - assert_equal(self.validate_zeroes(), 0) + self.assertEqual(self.nvme_write(), 0) + self.assertEqual(self.nvme_read(), 0) + self.assertEqual(self.validate_write_read(), 0) + self.assertEqual(self.write_zeroes(), 0) + self.assertEqual(self.nvme_read(), 0) + self.assertEqual(self.validate_zeroes(), 0) |