Apply changes from ruff

Author: ElliottKasoar

abcd/backends/atoms_opensearch.py

@@ -940,9 +940,7 @@ def delete_property(self, name: str, query: dict | str | None = None):
 
         self.client.update_by_query(index=self.index_name, body=body)
 
-    def hist(
-        self, name: str, query: dict | str | None = None, **kwargs
-    ) -> dict | None:
+    def hist(self, name: str, query: dict | str | None = None, **kwargs) -> dict | None:
         """
         Calculate histogram statistics for a property from all matching documents.
 

abcd/backends/atoms_properties.py

@@ -84,12 +84,12 @@ def __init__(
         self.encoding = encoding
         try:
             self.df = pd.read_csv(self.data_file, encoding=self.encoding)
-        except UnicodeDecodeError:
+        except UnicodeDecodeError as err:
             detected = chardet.detect(Path(self.data_file).read_bytes())
             raise ValueError(
                 f"File cannot be decoded using encoding: {self.encoding}."
                 f" Detected encoding: {detected}."
-            )
+            ) from err
         except pd.errors.ParserError:
             self.df = pd.read_excel(self.data_file, header=0)
 
@@ -154,11 +154,11 @@ def set_struct_files(self):
         for i in range(len(self.df)):
             try:
                 struct_name = self.df.iloc[i][self.struct_name_label]
-            except KeyError:
+            except KeyError as err:
                 raise ValueError(
                     f"{self.struct_name_label} is not a valid column in "
                     "the data loaded."
-                )
+                ) from err
             struct_file = self.get_struct_file(struct_name)
             self.struct_files.append(struct_file)
 

abcd/backends/utils.py

@@ -19,18 +19,18 @@ def histogram(name, data, **kwargs):
             print("Mixed type error of the %s property!", name)
             return None
 
-        if ptype == float:
+        if isinstance(data[0], float):
             bins = kwargs.get("bins", 10)
             return _hist_float(name, data, bins)
 
-        if ptype == int:
+        if isinstance(data[0], int):
             bins = kwargs.get("bins", 10)
             return _hist_int(name, data, bins)
 
-        if ptype == str:
+        if isinstance(data[0], str):
             return _hist_str(name, data, **kwargs)
 
-        if ptype == datetime:
+        if isinstance(data[0], datetime):
             bins = kwargs.get("bins", 10)
             return _hist_date(name, data, bins)
 

tests/test_opensearch_mock.py

@@ -56,7 +56,7 @@ def test_create(self, abcd):
         abcd.destroy()
         abcd.create()
         assert abcd.client.indices.exists("test_index") is True
-        abcd.client.indices.exists("fake_index") is False
+        assert abcd.client.indices.exists("fake_index") is False
 
     def test_push(self, abcd):
         """