import unittest
import sys
class TestDisk(unittest.TestCase):
@classmethod
def setUpClass(cls):
print("Disk Test Module Setup")
@classmethod
def tearDownClass(cls):
print("Disk Test Module Teardown")
def test_disk_info_01(self):
print("Running test_disk_info_01")
if __name__ == "__main__":
if len(sys.argv) > 1:
suite = unittest.TestSuite()
for test_name in sys.argv[1:]:
suite.addTest(TestCPU(test_name))
unittest.TextTestRunner().run(suite)
else:
unittest.main()