def test_check_clustersize_quotas_exceeded(self):
# arrange
prev_clustersize = self.opts['clustersize']
self.opts['clustersize'] = 3
c_yarn_cluster = YarnCluster(self.opts)
expected = error_quotas_clustersize # usage: 4, limit: 6 (2 remaining), requested: 3, expect error
# act
returned = c_yarn_cluster.check_clustersize_quotas()
self.opts['clustersize'] = prev_clustersize
#assert
self.assertEqual(expected, returned)
def test_check_clustersize_quotas_sufficient(self):
# arrange
prev_clustersize = self.opts['clustersize']
self.opts['clustersize'] = 2
c_yarn_cluster = YarnCluster(self.opts)
expected = 0 # usage: 4, limit: 6 (2 remaining), requested: 2, expected result success
# act
returned = c_yarn_cluster.check_clustersize_quotas()
self.opts['clustersize'] = prev_clustersize
#assert
self.assertEqual(expected, returned)
