summaryrefslogtreecommitdiff
path: root/morphlib/morphology.py
diff options
context:
space:
mode:
Diffstat (limited to 'morphlib/morphology.py')
-rw-r--r--morphlib/morphology.py12
1 files changed, 11 insertions, 1 deletions
diff --git a/morphlib/morphology.py b/morphlib/morphology.py
index 734c4e3d..f637f0f4 100644
--- a/morphlib/morphology.py
+++ b/morphlib/morphology.py
@@ -102,7 +102,17 @@ class Morphology(object):
@property
def disk_size(self):
- return self._dict['disk-size']
+ size = self._dict['disk-size']
+ size = size.lower()
+ if size.endswith('g'):
+ size = int(size[:-1]) * 1024**3
+ elif size.endswith('m'):
+ size = int(size[:-1]) * 1024**2
+ elif size.endswith('k'):
+ size = int(size[:-1]) * 1024
+ else:
+ size = int(size)
+ return size
@property
def test_stories(self):