Fixes #2 and generates empty uns if asked for

src/dummy_anndata/generate_dict.py

@@ -1,9 +1,8 @@
-from .generate_vector import vector_generators
-from .generate_matrix import matrix_generators
-
-import pandas as pd
 import numpy as np
 
+from .generate_matrix import matrix_generators
+from .generate_vector import vector_generators
+
 scalar_generators = {
     "string": "version",
     "char": "a",
@@ -34,17 +33,13 @@ def generate_type(type, n_rows, n_cols):
 
 def generate_dict(n_rows, n_cols, types=None, nested=True):
     if types is None:  # types are all vectors and all matrices
-        scalar_types = list(scalar_generators.keys()) + [
-            f"scalar_{t}" for t in vector_generators.keys()
-        ]
-        types = (
-            scalar_types
-            + list(vector_generators.keys())
-            + list(matrix_generators.keys())
-        )
-
-    data = {t: generate_type(t, n_rows, n_cols) for t in types}
-    if nested:
-        data["nested"] = generate_dict(n_rows, n_cols, types, False)
+        scalar_types = list(scalar_generators.keys()) + [f"scalar_{t}" for t in vector_generators.keys()]
+        types = scalar_types + list(vector_generators.keys()) + list(matrix_generators.keys())
+
+    data = {}
+    if types:  # types is not empty
+        data = {t: generate_type(t, n_rows, n_cols) for t in types}
+        if nested:
+            data["nested"] = generate_dict(n_rows, n_cols, types, False)
 
     return data

tests/test_basic.py

@@ -1,5 +1,3 @@
-import pytest
-
 import dummy_anndata
 
 
@@ -13,3 +11,9 @@ def test_generating_dataset(tmp_path):
     dummy = dummy_anndata.generate_dataset()
     filename = tmp_path / "dummy.h5ad"
     dummy.write_h5ad(filename)
+
+
+def test_empty_uns():
+    dummy = dummy_anndata.generate_dataset(uns_types=[])
+
+    assert dummy.uns == {}