def test_check_disk_valid_exceeded(self): # arrange prev_clustersize = self.opts['cluster_size'] self.opts['cluster_size'] = 100 c_yarn_cluster = YarnCluster(self.opts) expected = error_quotas_cyclades_disk # act with self.assertRaises(ClientError) as context: c_yarn_cluster.check_disk_valid() self.opts['cluster_size'] = prev_clustersize # assert the_exception = context.exception self.assertEqual(expected, the_exception.status)
def test_check_disk_valid_sufficient(self): # arrange c_yarn_cluster = YarnCluster(self.opts) expected = 0 # act returned = c_yarn_cluster.check_disk_valid() # assert self.assertEqual(expected, returned)