def test_get(self, lib): "Tests get() method." def pqos_mba_cos_num_mock(_p_cap, num_cos_ref): "Mock pqos_mba_get_cos_num()." ctypes_ref_set_int(num_cos_ref, 2) return 0 def pqos_mba_get_mock(socket, max_num_cos, num_cos_ref, cos_arr): "Mock pqos_mba_get()." self.assertEqual(socket, 1) self.assertEqual(max_num_cos, 2) self.assertEqual(len(cos_arr), max_num_cos) ctypes_ref_set_int(num_cos_ref, 2) cos_arr[0].class_id = 0 cos_arr[0].mb_max = 4000 cos_arr[0].ctrl = 1 cos_arr[1].class_id = 1 cos_arr[1].mb_max = 8000 cos_arr[1].ctrl = 1 return 0 lib.pqos_cap_get = MagicMock(return_value=0) lib.pqos_mba_get_cos_num = MagicMock(side_effect=pqos_mba_cos_num_mock) lib.pqos_mba_get = MagicMock(side_effect=pqos_mba_get_mock) mba = PqosMba() coses = mba.get(1) self.assertEqual(len(coses), 2) self.assertEqual(coses[0].class_id, 0) self.assertEqual(coses[0].mb_max, 4000) self.assertTrue(coses[0].ctrl) self.assertEqual(coses[1].class_id, 1) self.assertEqual(coses[1].mb_max, 8000) self.assertTrue(coses[1].ctrl) lib.pqos_mba_get.assert_called_once() lib.pqos_mba_get_cos_num.assert_called_once()
def test_get(self, lib): "Tests get() method." def pqos_mba_cos_num_mock(_p_cap, num_cos_ref): "Mock pqos_mba_get_cos_num()." ctypes_ref_set_int(num_cos_ref, 2) return 0 def pqos_mba_get_mock(socket, max_num_cos, num_cos_ref, cos_arr): "Mock pqos_mba_get()." self.assertEqual(socket, 1) self.assertEqual(max_num_cos, 2) self.assertEqual(len(cos_arr), max_num_cos) ctypes_ref_set_int(num_cos_ref, 2) cos_arr[0].class_id = 0 cos_arr[0].mb_max = 4000 cos_arr[0].ctrl = 1 cos_arr[1].class_id = 1 cos_arr[1].mb_max = 8000 cos_arr[1].ctrl = 1 return 0 lib.pqos_cap_get = MagicMock(return_value=0) lib.pqos_mba_get_cos_num = MagicMock(side_effect=pqos_mba_cos_num_mock) lib.pqos_mba_get = MagicMock(side_effect=pqos_mba_get_mock) mba = PqosMba() coses = mba.get(1) self.assertEqual(len(coses), 2) self.assertEqual(coses[0].class_id, 0) self.assertEqual(coses[0].mb_max, 4000) self.assertTrue(coses[0].ctrl) self.assertEqual(coses[1].class_id, 1) self.assertEqual(coses[1].mb_max, 8000) self.assertTrue(coses[1].ctrl) lib.pqos_mba_get.assert_called_once() lib.pqos_mba_get_cos_num.assert_called_once()
def print_allocation_config(sockets): """ Prints allocation configuration. Parameters: sockets: array with socket IDs """ mba = PqosMba() for socket in sockets: try: coses = mba.get(socket) print("MBA COS definitions for Socket %u:" % socket) for cos in coses: cos_params = (cos.class_id, cos.mb_max) print(" MBA COS%u => %u%% available" % cos_params) except: print("Error") raise
# Check if MBA is supported cap = PqosCap() mba_supported = False try: cap.get_type('mba') mba_supported = True except: pass print('Is MBA supported? %s' % ('Yes' if mba_supported else 'No')) if mba_supported: mba = PqosMba() # Get MBA configuration for socket 0 coses = mba.get(0) print('MBA configuration for socket 0:') for cos in coses: print('COS: %d' % cos.class_id) print('MB max: %d' % cos.mb_max) print('CTRL: %s' % cos.ctrl) print('') # Configure MBA, 50% of available bandwidth for COS 0 cos = mba.COS(0, 50) # Apply for socket 0 actual = mba.set(0, [cos]) print('Trying to set %d%%...' % cos.mb_max) print('Actually set: %d%%' % actual[0].mb_max)
# Check if MBA is supported cap = PqosCap() mba_supported = False try: cap.get_type('mba') mba_supported = True except: pass print('Is MBA supported? %s' % ('Yes' if mba_supported else 'No')) if mba_supported: mba = PqosMba() # Get MBA configuration for socket 0 coses = mba.get(0) print('MBA configuration for socket 0:') for cos in coses: print('COS: %d' % cos.class_id) print('MB max: %d' % cos.mb_max) print('CTRL: %s' % cos.ctrl) print('') # Configure MBA, 50% of available bandwidth for COS 0 cos = mba.COS(0, 50) # Apply for socket 0 actual = mba.set(0, [cos]) print('Trying to set %d%%...' % cos.mb_max) print('Actually set: %d%%' % actual[0].mb_max)