def test_check_cpu_valid_exceeded(self): # arrange prev_clustersize = self.opts['clustersize'] self.opts['clustersize'] = 3 c_yarn_cluster = YarnCluster(self.opts) expected = error_quotas_cpu # usage: 5, limit: 9 (4 remaining), requested: 2 + 4, expect result error # act returned = c_yarn_cluster.check_cpu_valid() self.opts['clustersize'] = prev_clustersize #assert self.assertEqual(expected, returned)
def test_check_cpu_valid_sufficient(self): # arrange prev_clustersize = self.opts['clustersize'] self.opts['clustersize'] = 2 c_yarn_cluster = YarnCluster(self.opts) expected = 0 # usage: 5, limit: 9 (4 remaining), requested: 4, expected result success # act returned = c_yarn_cluster.check_cpu_valid() self.opts['clustersize'] = prev_clustersize #assert self.assertEqual(expected, returned)