def test_check_cpu_valid_exceeded(self):
# arrange
prev_clustersize = self.opts['cluster_size']
self.opts['cluster_size'] = 30
c_yarn_cluster = YarnCluster(self.opts)
expected = error_quotas_cpu
# act
with self.assertRaises(ClientError) as context:
c_yarn_cluster.check_cpu_valid()
self.opts['cluster_size'] = prev_clustersize
# assert
the_exception = context.exception
self.assertEqual(expected, the_exception.status)
def test_check_cpu_valid_sufficient(self):
# arrange
prev_clustersize = self.opts['cluster_size']
self.opts['cluster_size'] = 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['cluster_size'] = prev_clustersize
# assert
self.assertEqual(expected, returned)
