Add support for specifying healpix order

Cosmoglobe · Apr 28, 2024 · a46cb20 · a46cb20
1 parent 53a1a1f
commit a46cb20
Show file tree

Hide file tree

Showing 5 changed files with 184 additions and 95 deletions.
diff --git a/tests/_strategies.py b/tests/_strategies.py
@@ -3,7 +3,7 @@
 import datetime
 from functools import partial
 from math import log2
-from typing import Any, Callable, Sequence
+from typing import Any, Callable, Literal, Sequence
 
 import astropy.coordinates as coords
 import astropy.units as u
@@ -71,8 +71,9 @@ def times(draw: DrawFn) -> time.Time:
 
 
 @composite
-def nsides(draw: Callable[[SearchStrategy[int]], int]) -> int:
-    return draw(integers(min_value=MIN_NSIDE_EXP, max_value=MAX_NSIDE_EXP).map(partial(pow, 2)))
+def healpixes(draw: DrawFn, order: Literal["ring", "nested"] = "ring") -> hp.HEALPix:
+    nside = draw(integers(min_value=MIN_NSIDE_EXP, max_value=MAX_NSIDE_EXP).map(partial(pow, 2)))
+    return hp.HEALPix(nside=nside, order=order)
 
 
 @composite