<s> """<STR_LIT>""" <EOL> import unittest <EOL> from perfkitbenchmarker import benchmark_spec <EOL> from perfkitbenchmarker import context <EOL> from perfkitbenchmarker import os_types <EOL> from perfkitbenchmarker . configs import benchmark_config_spec <EOL> from tests import mock_flags <EOL> _BENCHMARK_NAME = '<STR_LIT:name>' <EOL> _BENCHMARK_UID = '<STR_LIT>' <EOL> class _DiskTypeRenamingTestCase ( unittest . TestCase ) : <EOL> def setUp ( self ) : <EOL> self . mocked_flags = mock_flags . PatchTestCaseFlags ( self ) <EOL> self . mocked_flags . os_type = os_types . DEBIAN <EOL> self . addCleanup ( context . SetThreadBenchmarkSpec , None ) <EOL> def _CreateBenchmarkSpec ( self , config_dict ) : <EOL> config_spec = benchmark_config_spec . BenchmarkConfigSpec ( <EOL> _BENCHMARK_NAME , flag_values = self . mocked_flags , ** config_dict ) <EOL> spec = benchmark_spec . BenchmarkSpec ( config_spec , _BENCHMARK_NAME , <EOL> _BENCHMARK_UID ) <EOL> spec . ConstructVirtualMachines ( ) <EOL> return spec <EOL> class GcpDiskTypeRenamingTest ( _DiskTypeRenamingTestCase ) : <EOL> """<STR_LIT>""" <EOL> def testPDStandard ( self ) : <EOL> config = { <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> } <EOL> } , <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : <NUM_LIT:2> , <EOL> } <EOL> } <EOL> } <EOL> } <EOL> } <EOL> spec = self . _CreateBenchmarkSpec ( config ) <EOL> self . assertEquals ( spec . vms [ <NUM_LIT:0> ] . disk_specs [ <NUM_LIT:0> ] . disk_type , '<STR_LIT>' ) <EOL> def testPDSSD ( self ) : <EOL> config = { <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> } <EOL> } , <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : <NUM_LIT:2> , <EOL> } <EOL> } <EOL> } <EOL> } <EOL> } <EOL> spec = self . _CreateBenchmarkSpec ( config ) <EOL> self . assertEquals ( spec . vms [ <NUM_LIT:0> ] . disk_specs [ <NUM_LIT:0> ] . disk_type , '<STR_LIT>' ) <EOL> class AwsDiskTypeRenamingTest ( _DiskTypeRenamingTestCase ) : <EOL> def testEBSStandard ( self ) : <EOL> config = { <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : '<STR_LIT>' <EOL> } <EOL> } , <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : <NUM_LIT:2> <EOL> } <EOL> } <EOL> } <EOL> } <EOL> } <EOL> spec = self . _CreateBenchmarkSpec ( config ) <EOL> self . assertEquals ( spec . vms [ <NUM_LIT:0> ] . disk_specs [ <NUM_LIT:0> ] . disk_type , '<STR_LIT>' ) <EOL> def testEBSGP ( self ) : <EOL> config = { <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : '<STR_LIT>' <EOL> } <EOL> } , <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : <NUM_LIT:2> <EOL> } <EOL> } <EOL> } <EOL> } <EOL> } <EOL> spec = self . _CreateBenchmarkSpec ( config ) <EOL> self . assertEquals ( spec . vms [ <NUM_LIT:0> ] . disk_specs [ <NUM_LIT:0> ] . disk_type , '<STR_LIT>' ) <EOL> def testEBSPIOPS ( self ) : <EOL> config = { <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : '<STR_LIT>' <EOL> } <EOL> } , <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : <NUM_LIT:2> <EOL> } <EOL> } <EOL> } <EOL> } <EOL> } <EOL> spec = self . _CreateBenchmarkSpec ( config ) <EOL> self . assertEquals ( spec . vms [ <NUM_LIT:0> ] . disk_specs [ <NUM_LIT:0> ] . disk_type , '<STR_LIT>' ) </s>
