if single object is passed, it will be used as array

bahmutov · Feb 4, 2015 · 33fcc2a · 33fcc2a
1 parent ecdd84c
commit 33fcc2a
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/expensive-keys.js b/expensive-keys.js
@@ -44,7 +44,7 @@
     Object.keys(obj).forEach(function (key) {
       var val = obj[key];
       if (typeof val === 'number') {
-        result[key] = toSizeMB(val);
+        result[key] = toSizeMB(obj[key]);
       }
     });
     return result;
@@ -53,6 +53,9 @@
   function propertySizes(keys, items) {
     if (arguments.length === 1) {
       items = keys;
+      if (!Array.isArray(items) && typeof items === 'object') {
+        items = [items];
+      }
       if (!items.length) {
         return {};
       }