Fixed wrong unallocated space warning.

tests/test_partitions.py

@@ -92,12 +92,14 @@ def check_allocation(dev):
         size = int(blocks) * 512
         if name == root_dev:
             max_size = size
+            if root_dev.startswith('md'):
+                total_size += size
         else:
             total_size += size
     unallocated = max_size - total_size
     unallocated_gbytes = to_gbytes(unallocated)
     if unallocated_gbytes > 1:
-        print('Warning, %s GB are unallocated on %s.' % (unallocated_gbytes, root_dev))
+        print('Warning: %s%s GB are unallocated on %s.%s' % (YELLOW, unallocated_gbytes, root_dev, DEF))
         return False
     return True