feat(day 42): implement video storage calculator with unit validation and byte conversion

Author: JohnSunny21

FreeCodeCamp/CodingQ/VideoStorage.py

@@ -0,0 +1,47 @@
+"""
+Video Storage
+Given a video size, a unit for the video size, a hard drive capacity, and a unit for the hard drive, return the number of videos the hard drive can store using the following constraints:
+
+The unit for the video size can be bytes ("B"), kilobytes ("KB"), megabytes ("MB"), or gigabytes ("GB").
+If not given one of the video units above, return "Invalid video unit".
+The unit of the hard drive capacity can be gigabytes ("GB") or terabytes ("TB").
+If not given one of the hard drive units above, return "Invalid drive unit".
+Return the number of whole videos 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
+1 TB	1000 GB
+For example, given 500, "MB", 100, and "GB as arguments, determine how many 500 MB videos can fit on a 100 GB hard drive.
+"""
+
+
+def number_of_videos(video_size, video_unit, drive_size, drive_unit):
+
+    unit_map = {
+        "KB": 1000,
+        "MB": 1000 * 1000,
+        "GB": 1000 * 1000 * 1000
+    }
+
+    drive_map = {
+        "GB": 1000 * 1000 * 1000 ,
+        "TB": 1000 * 1000 * 1000 * 1000
+    }
+    
+    if video_unit not in unit_map:
+        return "Invalid video unit"
+    
+    if drive_unit not in drive_map:
+        return "Invalid drive unit"
+    
+    video_bytes = video_size * unit_map[video_unit]
+
+    drive_bytes = drive_size * drive_map[drive_unit]
+
+    return drive_bytes // video_bytes
+
+if __name__ == "__main__":
+    print(number_of_videos(500,'MB',100,"GB"))

FreeCodeCamp/CodingQ/VideoStorageTest.py

@@ -0,0 +1,23 @@
+import unittest
+from VideoStorage import number_of_videos
+
+class VideoStorageTest(unittest.TestCase):
+
+    def test1(self):
+        self.assertEqual(number_of_videos(500,"MB",100,"GB"),200)
+
+    def test2(self):
+        self.assertEqual(number_of_videos(2000,"B",1,"TB"),"Invalid video unit")
+
+    def test3(self):
+        self.assertEqual(number_of_videos(2000,"MB",100000,"MB"),"Invalid drive unit")
+
+    def test4(self):
+        self.assertEqual(number_of_videos(500000,"KB",2,"TB"),4000)
+
+    def test5(self):
+        self.assertEqual(number_of_videos(1.5,"GB",2.2,"TB"),1466)
+
+
+if __name__ == "__main__":
+    unittest.main()