more utils test cases -- untested

API/oursin/utils.py

@@ -120,15 +120,12 @@ def sanitize_drive_url(url):
         file_id = parts[index + 1] #extracting the file id
         download_link = f"https://drive.google.com/uc?id={file_id}"
         return download_link
+    elif "/edit#gid=" in url:
+        return url.replace("/edit#gid=", "/export?format=csv&gid=")
     else:
         raise Exception("The url is not a valid google drive url. Please make sure the url is a valid google drive url.")
 
-def sanitize_extension(file_name): #NOT VERIFIED
-    file_name = sanitize_string(file_name)
-    if file_name.endswith(".parquet") or file_name[-4]== ".":
-        return(file_name)
-    else:
-        raise Exception("The file name is not valid. Please make sure the file name is valid.")
+
 def sanitize_side(acronym, sided):
     if sided == "full":
         return acronym

API/tests/test_utils.py

@@ -23,6 +23,44 @@ def test_sanitize_vector3(self):
         self.assertRaises(Exception, urchin.utils.sanitize_vector3, ["one fish", "two fish", "red fish", "blue fish"])
         self.assertEqual(urchin.utils.sanitize_vector3(("cat", "in the", "hat")), ["cat", "in the", "hat"])
         self.assertEqual(urchin.utils.sanitize_vector3(["cat", "in the", "hat"]), ["cat", "in the", "hat"])
+
+    def test_sanitize_colors(self):
+        """Test sanitize_colors function"""
+        self.assertEqual(urchin.utils.sanitize_color('#FFA500'), '#FFA500')
+        self.assertEqual(urchin.utils.sanitize_color('00FF00'), '#00FF00')
+        self.assertEqual(urchin.utils.sanitize_color([0.5, 0.2, 0.8]), '#7F33CC')
+        self.assertEqual(urchin.utils.sanitize_color([128, 64, 255]), '#8040FF')
+        self.assertRaises(Exception,urchin.utils.sanitize_color([300, 50, 100]))
+        self.assertRaises(Exception,urchin.utils.sanitize_color(123))
+
+    def test_sanitize_float(self):
+        """Test sanitize_float function"""
+        self.assertEqual(urchin.utils.sanitize_float(3.14), 3.14)
+        self.assertEqual(urchin.utils.sanitize_float("2.718"), 2.718)
+        self.assertRaises(Exception,urchin.utils.sanitize_float("invalid"))
+        self.assertRaises(Exception, urchin.utils.sanitize_float([1, 2, 3]))
+        self.assertRaises(Exception, urchin.utils.sanitize_float({'value': 42}))
+        self.assertRaises(Exception, urchin.utils.sanitize_float(None))
+        self.assertRaises(Exception, urchin.utils.sanitize_float(True))
+
+    ## material?
+
+    def test_sanitize_lists(self):
+        """Test sanitize_lists function"""
+        self.assertEqual(urchin.utils.sanitize_list([1, 2, 3, 4]), [1, 2, 3, 4])
+        self.assertEqual(urchin.utils.sanitize_list(42, length=3), [42, 42, 42])
+        self.assertRaises(Exception,urchin.utils.sanitize_list(3.14))
+        self.assertRaises(Exception,urchin.utils.sanitize_list("string"))
+
+    def test_sanitize_string(self):
+        """Test sanitize_string function"""
+        self.assertEqual(urchin.utils.sanitize_string("string"), "string")
+        self.assertEqual(urchin.utils.sanitize_string(""), "")
+        self.assertRaises(Exception,urchin.utils.sanitize_string([1, 2, 3]))
+        self.assertRaises(Exception, urchin.utils.sanitize_string({'value': 42}))
+        self.assertRaises(Exception, urchin.utils.sanitize_string(None))
+        self.assertRaises(Exception, urchin.utils.sanitize_string(True))    
+
 
     def test_sanitize_drive_url(self):
         """Test sanitize_drive_url function"""
@@ -32,7 +70,9 @@ def test_sanitize_drive_url(self):
         self.assertRaises(Exception, urchin.utils.sanitize_drive_url, "test")
         self.assertRaises(Exception, urchin.utils.sanitize_drive_url,"https://drive.google.com/drive/folders/1vaHEXjG0M2jjlMceSvZhbOFCer4cdbR4")
 
-
-
-
-
+    def test_sanitize_side(self):
+        """Test sanitize_side function"""
+        self.assertEqual(urchin.utils.sanitize_side("VISp", "full"), "VISp")
+        self.assertEqual(urchin.utils.sanitize_side("VISp", "left"), "VISp-lh")
+        self.assertEqual(urchin.utils.sanitize_side("VISp", "right"), "VISp-rh")
+        self.assertRaises(Exception,urchin.utils.sanitize_side("VISp", "invalid"))