mem_as_is and ebs_size_as_is

tests/tibanna/unicorn/test_ec2_utils.py

@@ -243,7 +243,7 @@ def test_execution_mem_cpu():
     assert 'args' in unicorn_dict
     assert 'config' in unicorn_dict
     assert 'instance_type' in unicorn_dict['config']
-    assert unicorn_dict['config']['instance_type'] == 't3.micro'
+    assert unicorn_dict['config']['instance_type'] == 't3.small'
 
 
 def test_execution_benchmark():
@@ -265,7 +265,7 @@ def test_execution_benchmark():
     assert 'config' in unicorn_dict
     assert 'instance_type' in unicorn_dict['config']
     assert unicorn_dict['config']['instance_type'] == 't3.micro'
-    assert unicorn_dict['config']['ebs_size'] == 10
+    assert unicorn_dict['config']['ebs_size'] == 15
     # cleanup afterwards
     s3.delete_objects(Bucket='tibanna-output',
                       Delete={'Objects': [{'Key': randomstr}]})
@@ -397,7 +397,7 @@ def test_update_config_ebs_size():
     execution = Execution(input_dict)
     execution.input_size_in_bytes = execution.get_input_size_in_bytes()
     execution.update_config_ebs_size()
-    assert execution.cfg.ebs_size == 10
+    assert execution.cfg.ebs_size == 15
     # cleanup afterwards
     s3.delete_objects(Bucket='tibanna-output',
                       Delete={'Objects': [{'Key': randomstr}]})
@@ -415,7 +415,8 @@ def test_update_config_ebs_size2():
                            'app_name': 'md5',
                            'cwl_main_filename': 'md5.cwl',
                            'cwl_directory_url': 'someurl'},
-                  'config': {'log_bucket': 'tibanna-output', 'ebs_size': '5000000000x'}}
+                  'config': {'log_bucket': 'tibanna-output', 'ebs_size': '5000000000x',
+                             'ebs_size_as_is': True}}
     execution = Execution(input_dict)
     execution.input_size_in_bytes = execution.get_input_size_in_bytes()
     execution.update_config_ebs_size()
@@ -629,7 +630,7 @@ def test_launch_args():
     launch_args = execution.launch_args
     print(launch_args)
     assert launch_args
-    assert 't3.micro' in str(launch_args)
+    assert 't3.small' in str(launch_args)
     assert 'InstanceMarketOptions' in str(launch_args)
 
 
@@ -694,7 +695,7 @@ def test_ec2_exception_coordinator4():
                            'cwl_main_filename': 'md5.cwl',
                            'cwl_directory_url': 'someurl'},
                   'config': {'log_bucket': log_bucket, 'mem': 1, 'cpu': 1,
-                             'spot_instance': True,
+                             'spot_instance': True, 'mem_as_is': True,
                              'behavior_on_capacity_limit': 'other_instance_types'},
                   'jobid': jobid}
     execution = Execution(input_dict, dryrun=True)
@@ -779,7 +780,7 @@ def test_ec2_exception_coordinator8():
                            'cwl_main_filename': 'md5.cwl',
                            'cwl_directory_url': 'someurl'},
                   'config': {'log_bucket': log_bucket, 'instance_type': 't2.micro',
-                             'mem': 1, 'cpu': 1,
+                             'mem': 1, 'cpu': 1, 'mem_as_is': True,
                              'behavior_on_capacity_limit': 'other_instance_types'},
                   'jobid': jobid}
     execution = Execution(input_dict, dryrun=True)
@@ -803,7 +804,7 @@ def test_ec2_exception_coordinator9():
                            'cwl_main_filename': 'md5.cwl',
                            'cwl_directory_url': 'someurl'},
                   'config': {'log_bucket': log_bucket,
-                             'mem': 2, 'cpu': 1,
+                             'mem': 2, 'cpu': 1, 'mem_as_is': True,
                              'behavior_on_capacity_limit': 'other_instance_types'},
                   'jobid': jobid}
     execution = Execution(input_dict, dryrun=True)

tibanna/_version.py

@@ -1,4 +1,4 @@
 """Version information."""
 
 # The following line *must* be the last in the module, exactly as formatted:
-__version__ = "1.2.0b1"
+__version__ = "1.2.0b2"

tibanna/ec2_utils.py

@@ -301,6 +301,10 @@ def fill_default(self):
             self.root_ebs_size = DEFAULT_ROOT_EBS_SIZE
         if not hasattr(self, 'awsf_image'):
             self.awsf_image = DEFAULT_AWSF_IMAGE
+        if not hasattr(self, 'mem_as_is'):  # if false, add 1GB overhead
+            self.mem_as_is = False
+        if not hasattr(self, 'ebs_size_as_is'):  # if false, add 5GB overhead
+            self.ebs_size_as_is = False
 
     def fill_internal(self):
         # fill internally-used fields (users cannot specify these fields)
@@ -373,7 +377,11 @@ def init_instance_type_list(self):
                                             'EBS_optimized': False})
         # user specified mem and cpu
         if self.cfg.mem and self.cfg.cpu:
-            list0 = get_instance_types(self.cfg.cpu, self.cfg.mem, instance_list(exclude_t=False))
+            if self.cfg.mem_as_is:
+                mem = self.cfg.mem
+            else:
+                mem = self.cfg.mem + 1
+            list0 = get_instance_types(self.cfg.cpu, mem, instance_list(exclude_t=False))
             nonredundant_list = [i for i in list0 if i['instance_type'] != instance_type]
             instance_type_dlist.extend(nonredundant_list)
         # user specifically wanted EBS_optimized instances
@@ -421,7 +429,7 @@ def total_input_size_in_gb(self):
         except:
             return None
 
-    def auto_calculate_ebs_size(self):
+    def update_config_ebs_size(self):
         """if ebs_size is in the format of e.g. '3x', it updates the size
         to be total input size times three. If the value is lower than 10GB,
         keep 10GB"""
@@ -436,13 +444,13 @@ def auto_calculate_ebs_size(self):
                 self.cfg.ebs_size = round(self.cfg.ebs_size) + 1
             else:
                 self.cfg.ebs_size = round(self.cfg.ebs_size)
-            if self.cfg.ebs_size < 10:
-                self.cfg.ebs_size = 10
-
-    def update_config_ebs_size(self):
-        self.auto_calculate_ebs_size()  # if in the format of '3x'
         if not self.user_specified_ebs_size:  # use benchmark only if not set by user
             self.cfg.ebs_size = self.benchmark['ebs_size']
+        if not self.cfg.ebs_size_as_is:
+            self.cfg.ebs_size += 5  # account for docker image size
+        if self.cfg.ebs_size < 10:
+            self.cfg.ebs_size = 10
+
 
     def get_input_size_in_bytes(self):
         input_size_in_bytes = dict()