feat(day 41): implement file-storage capacity calculator with unit conversion

Author: JohnSunny21

FreeCodeCamp/CodingQ/FileStorage2.py

@@ -0,0 +1,39 @@
+"""
+File Storage
+Given a file size, a unit for the file size, and hard drive capacity in gigabytes (GB), return the number of files the hard drive can store using the following constraints:
+
+The unit for the file size can be bytes ("B"), kilobytes ("KB"), or megabytes ("MB").
+Return the number of whole files the drive can fit.
+Use the following conversions:
+Unit	Equivalent
+1 B	1 B
+1 KB	1000 B
+1 MB	1000 KB
+1 GB	1000 MB
+For example, given 500, "KB", and 1 as arguments, determine how many 500 KB files can fit on a 1 GB hard drive.
+
+
+
+"""
+
+def number_of_files(file_size, file_unit, drive_size_gb):
+
+    if file_unit == "B":
+        size_in_bytes = file_size
+    elif file_unit == "KB":
+        size_in_bytes = file_size * 1000
+    elif file_unit == "MB":
+        size_in_bytes = file_size * 1000 * 1000
+    else:
+        raise ValueError("Invalid unit. Use 'B', 'KB', or 'MB'.")
+    
+
+    drive_size_bytes = drive_size_gb * 1000 * 1000 * 1000
+    return drive_size_bytes // size_in_bytes
+
+
+
+
+if __name__ == "__main__":
+    print(number_of_files(500,"KB",1))
+    print(number_of_files(50000,"B",1))

FreeCodeCamp/CodingQ/FileStorage2Test.py

@@ -0,0 +1,26 @@
+import unittest
+from FileStorage2 import number_of_files
+
+class FileStorge2Test(unittest.TestCase):
+
+    def test1(self):
+        self.assertEqual(number_of_files(500,"KB",1),2000)
+    
+    def test2(self):
+        self.assertEqual(number_of_files(50000,"B",1),20000)
+
+    def test3(self):
+        self.assertEqual(number_of_files(5,"MB",1),200)
+
+    def test4(self):
+        self.assertEqual(number_of_files(4096,"B",1.5),366210)
+
+    def test5(self):
+        self.assertEqual(number_of_files(220.5,"KB",100),453514)
+
+    def test6(self):
+        self.assertEqual(number_of_files(4.5,"MB",750),166666)
+
+
+if __name__ == "__main__":
+    unittest.main()