From c39621d6186b4c47df3d2736592bc9684db5c756 Mon Sep 17 00:00:00 2001
From: Melissa DeLucchi <113376043+delucchi-cmu@users.noreply.github.com>
Date: Thu, 21 Nov 2024 13:09:34 -0500
Subject: [PATCH] Create new catalog type: map (#429)

* Initial work on map type.

* Use importer to create map type.
---
 src/hats/catalog/__init__.py                  |   1 +
 src/hats/catalog/catalog_type.py              |   1 +
 src/hats/catalog/dataset/table_properties.py  |   2 +
 src/hats/catalog/map/__init__.py              |   0
 src/hats/catalog/map/map_catalog.py           |   7 +++
 src/hats/loaders/read_hats.py                 |   4 +-
 tests/data/generate_data.ipynb                |  47 +++++++++++++++++-
 .../dataset/Norder=1/Dir=0/Npix=0.parquet     | Bin 0 -> 2881 bytes
 .../dataset/Norder=1/Dir=0/Npix=12.parquet    | Bin 0 -> 2881 bytes
 .../dataset/Norder=1/Dir=0/Npix=16.parquet    | Bin 0 -> 2881 bytes
 .../dataset/Norder=1/Dir=0/Npix=20.parquet    | Bin 0 -> 2881 bytes
 .../dataset/Norder=1/Dir=0/Npix=24.parquet    | Bin 0 -> 2881 bytes
 .../dataset/Norder=1/Dir=0/Npix=28.parquet    | Bin 0 -> 2881 bytes
 .../dataset/Norder=1/Dir=0/Npix=32.parquet    | Bin 0 -> 2881 bytes
 .../dataset/Norder=1/Dir=0/Npix=36.parquet    | Bin 0 -> 2881 bytes
 .../dataset/Norder=1/Dir=0/Npix=4.parquet     | Bin 0 -> 2881 bytes
 .../dataset/Norder=1/Dir=0/Npix=40.parquet    | Bin 0 -> 2881 bytes
 .../dataset/Norder=1/Dir=0/Npix=44.parquet    | Bin 0 -> 2881 bytes
 .../dataset/Norder=1/Dir=0/Npix=8.parquet     | Bin 0 -> 2881 bytes
 .../data/square_map/dataset/_common_metadata  | Bin 0 -> 1612 bytes
 tests/data/square_map/dataset/_metadata       | Bin 0 -> 8773 bytes
 tests/data/square_map/partition_info.csv      |  13 +++++
 tests/data/square_map/point_map.fits          | Bin 0 -> 8640 bytes
 tests/data/square_map/properties              |  12 +++++
 tests/hats/catalog/loaders/test_read_hats.py  |   2 +
 25 files changed, 86 insertions(+), 3 deletions(-)
 create mode 100644 src/hats/catalog/map/__init__.py
 create mode 100644 src/hats/catalog/map/map_catalog.py
 create mode 100644 tests/data/square_map/dataset/Norder=1/Dir=0/Npix=0.parquet
 create mode 100644 tests/data/square_map/dataset/Norder=1/Dir=0/Npix=12.parquet
 create mode 100644 tests/data/square_map/dataset/Norder=1/Dir=0/Npix=16.parquet
 create mode 100644 tests/data/square_map/dataset/Norder=1/Dir=0/Npix=20.parquet
 create mode 100644 tests/data/square_map/dataset/Norder=1/Dir=0/Npix=24.parquet
 create mode 100644 tests/data/square_map/dataset/Norder=1/Dir=0/Npix=28.parquet
 create mode 100644 tests/data/square_map/dataset/Norder=1/Dir=0/Npix=32.parquet
 create mode 100644 tests/data/square_map/dataset/Norder=1/Dir=0/Npix=36.parquet
 create mode 100644 tests/data/square_map/dataset/Norder=1/Dir=0/Npix=4.parquet
 create mode 100644 tests/data/square_map/dataset/Norder=1/Dir=0/Npix=40.parquet
 create mode 100644 tests/data/square_map/dataset/Norder=1/Dir=0/Npix=44.parquet
 create mode 100644 tests/data/square_map/dataset/Norder=1/Dir=0/Npix=8.parquet
 create mode 100644 tests/data/square_map/dataset/_common_metadata
 create mode 100644 tests/data/square_map/dataset/_metadata
 create mode 100644 tests/data/square_map/partition_info.csv
 create mode 100644 tests/data/square_map/point_map.fits
 create mode 100644 tests/data/square_map/properties

diff --git a/src/hats/catalog/__init__.py b/src/hats/catalog/__init__.py
index ee69f91f..5897d917 100644
--- a/src/hats/catalog/__init__.py
+++ b/src/hats/catalog/__init__.py
@@ -6,5 +6,6 @@
 from .catalog_type import CatalogType
 from .dataset.dataset import Dataset
 from .dataset.table_properties import TableProperties
+from .map.map_catalog import MapCatalog
 from .margin_cache.margin_catalog import MarginCatalog
 from .partition_info import PartitionInfo
diff --git a/src/hats/catalog/catalog_type.py b/src/hats/catalog/catalog_type.py
index 61ac8fb3..be36757a 100644
--- a/src/hats/catalog/catalog_type.py
+++ b/src/hats/catalog/catalog_type.py
@@ -9,6 +9,7 @@ class CatalogType(str, Enum):
     ASSOCIATION = "association"
     INDEX = "index"
     MARGIN = "margin"
+    MAP = "map"
 
     @classmethod
     def all_types(cls):
diff --git a/src/hats/catalog/dataset/table_properties.py b/src/hats/catalog/dataset/table_properties.py
index 94542eed..008f9500 100644
--- a/src/hats/catalog/dataset/table_properties.py
+++ b/src/hats/catalog/dataset/table_properties.py
@@ -25,6 +25,7 @@
     ],
     CatalogType.INDEX: ["primary_catalog", "indexing_column", "extra_columns"],
     CatalogType.MARGIN: ["primary_catalog", "margin_threshold", "ra_column", "dec_column", "default_columns"],
+    CatalogType.MAP: ["default_columns"],
 }
 
 ## catalog_name, catalog_type, and total_rows are required for ALL types
@@ -40,6 +41,7 @@
     ],
     CatalogType.INDEX: ["primary_catalog", "indexing_column"],
     CatalogType.MARGIN: ["primary_catalog", "margin_threshold"],
+    CatalogType.MAP: [],
 }
 
 # All additional properties in the HATS recommendation.
diff --git a/src/hats/catalog/map/__init__.py b/src/hats/catalog/map/__init__.py
new file mode 100644
index 00000000..e69de29b
diff --git a/src/hats/catalog/map/map_catalog.py b/src/hats/catalog/map/map_catalog.py
new file mode 100644
index 00000000..32cb6f15
--- /dev/null
+++ b/src/hats/catalog/map/map_catalog.py
@@ -0,0 +1,7 @@
+from __future__ import annotations
+
+from hats.catalog.healpix_dataset.healpix_dataset import HealpixDataset
+
+
+class MapCatalog(HealpixDataset):
+    """A HATS table to represent non-point-source data in a continuous map."""
diff --git a/src/hats/loaders/read_hats.py b/src/hats/loaders/read_hats.py
index f071e2d2..532f35cb 100644
--- a/src/hats/loaders/read_hats.py
+++ b/src/hats/loaders/read_hats.py
@@ -9,7 +9,7 @@
 from upath import UPath
 
 import hats.pixel_math.healpix_shim as hp
-from hats.catalog import AssociationCatalog, Catalog, CatalogType, Dataset, MarginCatalog
+from hats.catalog import AssociationCatalog, Catalog, CatalogType, Dataset, MapCatalog, MarginCatalog
 from hats.catalog.association_catalog.partition_join_info import PartitionJoinInfo
 from hats.catalog.dataset.table_properties import TableProperties
 from hats.catalog.index.index_catalog import IndexCatalog
@@ -23,6 +23,7 @@
     CatalogType.ASSOCIATION: AssociationCatalog,
     CatalogType.INDEX: IndexCatalog,
     CatalogType.MARGIN: MarginCatalog,
+    CatalogType.MAP: MapCatalog,
 }
 
 
@@ -63,6 +64,7 @@ def _is_healpix_dataset(dataset_type):
         CatalogType.SOURCE,
         CatalogType.ASSOCIATION,
         CatalogType.MARGIN,
+        CatalogType.MAP,
     )
 
 
diff --git a/tests/data/generate_data.ipynb b/tests/data/generate_data.ipynb
index 242e4bc7..3e1c9032 100644
--- a/tests/data/generate_data.ipynb
+++ b/tests/data/generate_data.ipynb
@@ -297,6 +297,49 @@
     "    runner.pipeline_with_client(args, client)"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import pandas as pd\n",
+    "import numpy as np\n",
+    "from hats.pixel_math.spatial_index import healpix_to_spatial_index\n",
+    "\n",
+    "target_pixels = np.arange(0, 12)\n",
+    "\n",
+    "healpix_29 = healpix_to_spatial_index(0, target_pixels)\n",
+    "\n",
+    "square_vals = target_pixels * target_pixels\n",
+    "value_frame = pd.DataFrame({\"_healpix_29\": healpix_29, \"star_count\": square_vals})"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "with tempfile.TemporaryDirectory() as pipeline_tmp:\n",
+    "    csv_file = Path(pipeline_tmp) / \"square_map.csv\"\n",
+    "    value_frame.to_csv(csv_file, index=False)\n",
+    "    args = ImportArguments(\n",
+    "        constant_healpix_order=1,\n",
+    "        catalog_type=\"map\",\n",
+    "        use_healpix_29=True,\n",
+    "        ra_column=None,\n",
+    "        dec_column=None,\n",
+    "        file_reader=\"csv\",\n",
+    "        input_file_list=[csv_file],\n",
+    "        output_artifact_name=\"square_map\",\n",
+    "        output_path=\".\",\n",
+    "        tmp_dir=pipeline_tmp,\n",
+    "    )\n",
+    "\n",
+    "    runner.pipeline_with_client(args, client)"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,
@@ -317,7 +360,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "env",
+   "display_name": "demo",
    "language": "python",
    "name": "python3"
   },
@@ -331,7 +374,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.9.12"
+   "version": "3.12.3"
   }
  },
  "nbformat": 4,
diff --git a/tests/data/square_map/dataset/Norder=1/Dir=0/Npix=0.parquet b/tests/data/square_map/dataset/Norder=1/Dir=0/Npix=0.parquet
new file mode 100644
index 0000000000000000000000000000000000000000..d51cc17aec77ce710c9f55a0b13022ecb627ce66
GIT binary patch
literal 2881
zcmcgu&u<z>6du-y5ZebU&8|S=fRMb(!Ephzj>)OB81NDs+gM?Nrb7G6u-5KEcEPY%
zk#eefta|Dx=N@|KvB#)~s^=c6{uMp8*S?uu2ka!O(%5-|VdlLz?|tvhHzSU4BgaPB
zHTM2K%dqGkLh%#|g+=T=mfHaR8XIF{sYLwJGM{ni!8^3zcQ?(z$V4Ix;(v>z*l1Ef
zv9)AkrQ0_YYiLfp`6A134EJ+{dmZ6-`1ps!Pi_Ec#4-qmHTXg3){FQ(Ixk|`8!2vA
z8;qUNLJhwx@!u_dbQN)S_3|j9s}WM!wPV;u528e$pd(BcTm9-zie<u9$La%DH(ZKv
z%Ds&6zsC4ivDCHw|G$+G=)ceY_1$^fmUhhm&9@Qm`w0JMoc|$C;Jho!EyMb2WezJ^
z2RhO&1-QOm;s03v5L`)1K#7>A^$lBLmw}4w%oUyong!ATbHOD%pLK5k$Is$$^8z>0
zr1cc2zIdae0OvZ6Nv4v|KFO`(KMCvkn_w+}b4=rJkR9ANF^=Km+)uIl2qiZXv7zGV
zikJ9@xyYD~ZcO1BvBtLJWge%${Awee36@=IGp=dr@*?9XwgK*$g{L$FJ5No+(!1|5
z25Gw^e@M5Q&S<+pHaTN^=)bMDjgg`&BL&hpW6OH6k)EfWKYnj#y=wc&q_iEbp^OGD
z;JC<Grs^uLPf8#9it7$01i-oBdgj0(+T5ef>}ECtXft8z1!I0QzqyrpffB2@(KwVJ
z2lqAcGX6Q0UI}A-3|}HoFfL%RhbtKOaH&UiqIktVA<FU_`;!Tn$r^-4p>_%52?hiF
zk|3oru@OhSr@@c-NV2VQr`jyE^7V6duQ^hk1M|=n)z-<RC)uISDh^e<VHwrtOgbC%
z<XqO1O0_dJKh3GK5a_l%Q?)Cj&WW@Qwo|3r?5lg?(5USAhbI#8w-veRbt=ViTh3V|
zj${=F9l7FZRjbgNNrGfTPDkuRuD+HlW{s8@_yI<Jx1FVSTga&`QD|*71g-iMuwqk@
zg>3!o*xWB+Q<a<9w%q9J)iSALtjkmH6lyX2+B~mQ&$e5&({{d?Q=MR6$F?lB+BjXP
zVWw9rKIw}N@ox=VGw7veHvx0eCw&2|o+8(Lz$nl@Iidh_U)|DZjk?|RwY+t%nxxm#
zwC)!s`)AlBbuRQT-)Yrmid-Di7-!P>z%0xpCrfw*=O6YNXd)8691>EL+GM1|E+rA=
zQ{xB-O6YVfQDv1ly`gHzr?N0nS^-g?pM!sv_=poiZ8E`qvc^xzK>0@|81IsyQMG(r
z-v`ekSz`NIJ|Hg$O-*MMe_1%!fRkE~5+L<RTndd`Bz!P~csnpD1L;A;%lkqWL<aB`
zubeT|zZ2pOkwQ{v4#b1FS^yT31eXl9hyAFLa0~cKBvs`qw+TKd7D~}byinkJgP=ru
nsJPF^#^|9o96qEEOFDcw(!rCGd<OrD7eDClI|v=YpXYx8Z2M4y

literal 0
HcmV?d00001

diff --git a/tests/data/square_map/dataset/Norder=1/Dir=0/Npix=12.parquet b/tests/data/square_map/dataset/Norder=1/Dir=0/Npix=12.parquet
new file mode 100644
index 0000000000000000000000000000000000000000..657359aabf1e63c0acea975e5fce306035472931
GIT binary patch
literal 2881
zcmcgu&2QpH6gOVPIv<B<yS6~$fRJMCA)5lh*|4XM2@tahNr)B$-3s|D;K(+tfq;ot
z+EdkI)l*M<?xBYsdtCKU_1r_%zoN(X+BahpKEkT1n(h+}Gw;25@AuyPX2cn)rr9vN
z#Xde}X_k6KQB0ijWs{}YM=ZAk`YkrXM&eOsYMISA>d6W%_^l;q80k=mrubh%aW)(i
zsK{0<y3y(wiq$j6txS%kIhy-1#Jvyk2R!p{;>UIXG{w>s3|sI+QOiZF1z7JEF||aQ
z=?oV!ozJvOMf@6`6|vqQC~iyZ4xPbV4L?2Lzj**v+zs`C1S_jx*v;utgf~ODvZZ6#
zr=EndK1#+Vl<ywKSvqKSq|$YD!zBpE+}jZUbA*2vi7)M631Sr=RsorwJ$1_nz_W1n
z@7HE+dvIv_Xub|{--h_#8U8y4!&w#OpJ8p{R3Iz*!%S__+Zi~{Q}U0E8LV&x=t#Qc
z<N9)g|81S%S`lUyu9ziI(MVtG7`DQ$0~NRF1)lqwInn@g!6i7KZSMH{PZ+p)z8h}R
zb{tfn-Kq${nU14l@z|@6a;x}{!g~HbSj*oZ)A)O22lrKkqj`q=A@Z1_Vmr}DPjPgm
zAN`BINt%vsjNuuvhPKmBK2Lo9#ZDsWFI&WBENRK|Cg~`)0q)7Ur!)yWFHOVJTOTm`
zX<GxYN4A>IU_XmDIYYbW-BsJhK+%<f0%@F~Wxd%+%+k&tzq?r%ZT~STX@_elgRToW
zZjzR%x{B-J(tDoby4?{5aBaAKv+H1O`q^%3H<bjm$sqNdF|(W5-AlfqqMN8%J(ZvP
z_ci*K`IJaIj}SVCFP0YwWs%rMMTFWY-zGXyT(FO@vT$pEF#<DQgODi1E+I5RV1Qp7
zBwxTb?1+vu_z|B;wl!>)YT0_Ga;+ZK2C8#no|>XszZkV8JJ4CVo@!Suqg0znSKYRp
zPW7dH`AW@<)2b}^y1l-s+QmWhLfQx0rBbSO)FZKH6c4=93kmz%id^eAi@9M#PFpyR
zWaYX|x!BiAR<=Ho1j&S)j@W@*9W9+p8FkV31C07%BSq}CkXGxWP~WQxTInTVMW!MP
zsmj&4d7MY4D%VmCx!Tc71zg8am&g4}sKxNgv%F3v)u@**8<|{Mb^Luz+p^SB^>VI;
ziC!vtxGy@y|KG6H{9bBy4KU|C+!w&wSLCt>7zNTNM-*W0se2l!QMYTJma(o?6Zcvg
zSG?@#_zIc0&bj_&n)UKTk#j>5<4PKynAwTsq%g1G{DVFNO+?I>Q%s6{1CL}lBqSny
zs-6KsF`dpOqAX#jTdD?pDhMN`?i2O(HTb8nk2u2A#uL=RYxEKignxK~&><eGCCfvV
zWAM!3C9;?D0eL}4YBD4E3&OPqoYef30I7}R5@`4$p_2*3JAg?6NDm@T?+adF>B3iB
zIAf^)Aix_Sg{06NhzD^sA1oyCFBxnP`ccN==J4fls^TKIF+M03O3`q<K;U|VP|-wB
mabFLO!Bef*drBUbMDTDV{3j*$3jP((f2cnnQq&pzdHx#-rcvks

literal 0
HcmV?d00001

diff --git a/tests/data/square_map/dataset/Norder=1/Dir=0/Npix=16.parquet b/tests/data/square_map/dataset/Norder=1/Dir=0/Npix=16.parquet
new file mode 100644
index 0000000000000000000000000000000000000000..ccb88d01bb496cc6e999a3af7a7a452cd51d45ca
GIT binary patch
literal 2881
zcmcgu&2Jh<6kpbe5ZecoW>+9_KuBKY;JAQU$K=#m40wr+ZLF|BQ=$C=*4kYN3x>st
zlvCAX)l*M7_s~O+Jw`oLJ@-)cujsM8_RZ{KK1@_qbu)rt=Dj!X{oZ@O8F7xPX(q@-
zn8zm!%}|dhDjcW05n(9i5yS3+Ji>&SP&^u*Y33`Ida_0eb`KLY%yb|?Q`|3sI1`NV
zR45XQZne6)Z1#;wE0bermS%qpu<rxhAs7BP{$slUnqp`QrU?8{)M^qDAM4#DW}0Yt
zw!%rw)^q7f62AuLNo+bpdC*dOBYU`z!%rLBHye<}T~{9nur`Y@vps7?a65oATWP}q
z^(2VpQ8KTfeD@&E(0-{ymEJ(p2L$1SeH-9@4sq{7@s;yyL9FA$Iv}&&Q@4x&JWIWQ
z|8QQmjU&TD^L2pzHo*NJ=DrJKIP1LpGpt#e*VJcC3szwLu{DPktN<NJl{{QuZgIbD
z5?r$o^PW=caK%iXiiRAut6MU&2~^ynmw4_;7Dxlk1rxtNJM78#pM>G&d1kmsJ8=+w
zb}J$P=Q5U##bd8N%B|u)3hTN1U@dolOylm69qd;jmgd6j5242tkS7}I%eE#v(ZA@M
zq+x6N1fCIdWZ6#gdE)agb`wc&+afxBMM}0eNn5sbFi$Q#rAav1Hgr>KeL(4jZ4KQ%
zIcnI$gDgH|kF36XS8VA+S(Aq{gt14a`DQmU4?BPS?nYhK{l}oB90Oe*_6C6CCTSYV
zKpwa_^}Z_)2E8!`aIFs<qi17j`q^G;FO>weNk8<0GP9T2+fTlsqT8riJ(He$_ci)9
z{3#ioJVNLKzIePqD2s#+Dk9WD`3{i@!jgWBg@s%Fi!rG29tQD3^ddrI1Pb`YPVxn;
z!-nWsg&*OWXqlsCsg|v0D%Z+!ZK&9%#+e}~^~-Tbw0xPF>nm2()JwIgc-8Aj>69bp
z%U4Qfl2#<%lkGc(VikwYOYs17ZMjtID#t=!FCMyQmm;>eWU1yfi@8xlN}Je^Xy$rN
zspzOBGh3gEyl6m7Tj)ZpuA0uJ^t#~L0Y>epks^AFPb+nSukTlRwe%9OB14w=RORZz
zILRYJk!q=iRPAb|0?uQkNfW0Hx#(_r9@nm<8ufC!k;$bM+dJ2^Oj9XU+Y32NwNlZ=
zb<x28|AwvR)l#);fH~*lx&T&3mdY+*<Vl@ufrquL?5iY4&8oR-#=KSxTx)SsakJx-
zD`en27wVU3*2_~_%8iJRD{*vcWT&E?!n}h1`*jAI2$(Ntm=yU2p2>7XNJRKlJqLnf
zI$elFSi(lPRQ35(;Ky>^BkJpGuuowdVT`Gb7pRN(=p~*A|L_8#BRo}0ri&^kV41^P
zWUa&l^1>jY$%^1F@YgDEQuRUtqz?8=py7*zPN(4S5Eca>Jq%%XU+@Nx9(;wR9z*_z
zK3*Rw1cl;&Kk%!1U?GTi$>6wOk1}?*fG>|j6_>e<@j<eXii-XD1Fkm=Dw^oagV!T{
i_*CuppOS|q;XfP+?@5Whf`7$}AL`Es6m<@Np8p0HxKfh<

literal 0
HcmV?d00001

diff --git a/tests/data/square_map/dataset/Norder=1/Dir=0/Npix=20.parquet b/tests/data/square_map/dataset/Norder=1/Dir=0/Npix=20.parquet
new file mode 100644
index 0000000000000000000000000000000000000000..73ee41905ce974c141f1b282e36d1fe1029130a8
GIT binary patch
literal 2881
zcmcgu&2QpH6gOVPdXpTYZEP8d144?mhinQ2XTzR4CP2(4Bq3T1bSvbqfFs+md<0Ci
z(w?dwtDbt=a}PcA*yE~)s^=c6{uMp8*S;B>@DWy3)pVa=n0fEbd%ySQHzUqaCB=r>
zO?K;;rCI7RMKLkTmo1iJAG6#J=r`F28;SAE)H0iK)RPrj@Ou=eVWdMLni75q#n^CE
zq#~P9e!bZ?Rl94Ao9Qe|b2Rs3h<hIr4g}`k#E<R(Xo{sN7&hUDqLzz@23YSFF}3i_
zbcTzV&S%=CB7P0eidgFp)Lv8X4Bf$84L>~)zIgyu+zs`C1S_jx*p2B?gf~LCvZZ6#
zr=Endo+slH%6AWAEFH8uQttE&vqun)xwj$V=ZNqw5?k8862vM#tO7DUd+L@EfM?<C
z-#?nQ?ZKhtqxm|-eH#*fXN2z<3};o8e}=V%Q-Q4L4>Ps#w=-~@r{o{&Gg#p=(2;b>
z$Mxm9@Y@=}wIa+aTv1!3_()%Gn~ut^0Ts9D1)lqwInn@g!6i7KZSMH{PZ+p)z8h}R
zb_`UX-Kq${nU15QvFNLha;t=o!g}F8SS#Eg(}a6u2lrKkqXmZhAp&HI?(mVW>KbaF
z|BJp!Sgv7?;Tf@qj@wT>kAMEfPCVf+o5W@=Y02^?;i`@a?uoglGyyv=Ez>rdA29l9
zn**;)wp#9BKZ7^9L#ONARXgTDHPnF$Y22Z0zuAe;(#{^gyIB`)|1l|PN6%CTogU!0
zN!XUwQ+pmRz3ZvHUT1^>T${bV)p4;l^=vn}n@j-OM38#UnBGnA?j_z({06F2PL=2W
zedXUWpAu>15JKnh#qt8743b)?fKUtNT0|#F3-%FK=5OsUMqtKk5E6yhWrRiu4DgGC
z<nq{t9np~vKhiVVv4@RfHB(EMueGD<Kyy#5Q%lln7o(Q!1Uft0)triL7ONBas?$<Z
z$-bN`U1{lYN>fB%x7W8cr!Z(-$opWsREyQNb|iJp!hv^sA!C0>RjU0)Av>%qDI3R;
z?QEx^6#9D6&eSHdC|i)zmD-T2t*5d{vnKg|fYCUtCyCt=Q(8?DYkL(@FTMn<$Wj$C
zS-v{Aj&sP;lxniBRN6){kLws3%D8_CwU}OMme(yO>$TElJ)KQyuD`G0*tS-zT+Y=n
zF^UBb_r-wt{~NZd-%H)80_LoT`vO?|s#5X*qe%MXN+QfXZBHjP8cx;I)AqGy;a<z*
zvX>bhUm*+EIoH2*qgI-zN_I$MT*<=|D>ISZB<2;If6!;3iG=xbib;{H<B<%9ghYf-
zl`|kHrqj7hltt`xOVxl+d2yuHe4@U-2LB}Xkw%!>c!Jt^jb7q`@DEQAI>bYzXnUx9
z44zrMM9xw^ATJ0>O=bjtUcA<Ule(W0AhmE@0u5gzbTWZ>2QbM4=|QCFeZdPX9r#KM
zXAJcp1b73akQAB&@gT15gM}piC4=ojKT0^<9KIY*RaoRU#s|ehDLRf92wZOvijQ~I
m-s_<`c&c~1Pszg)4<3%V|D;4;!N2195B29miaLWo&wm3Fxl>pG

literal 0
HcmV?d00001

diff --git a/tests/data/square_map/dataset/Norder=1/Dir=0/Npix=24.parquet b/tests/data/square_map/dataset/Norder=1/Dir=0/Npix=24.parquet
new file mode 100644
index 0000000000000000000000000000000000000000..e98d36a6bdf0156d79775b542b2f77bf68e6e70c
GIT binary patch
literal 2881
zcmcgu&u`*J6gCcGy-5zyc5NAn1450phinQ2XTzR4CP2(4Bw@7}=vK%!4mh$6iy>g5
zmG)HiSoPG?o_pw_#~xQbR6X}l^{?o$z4p!6gdbs5t(xu=3^VV&dGC8~z8P_bDk(P1
zZm|!KS(>FDP!toReBoJ&eZX=%px<I6Y$V1rQ_F0|QIA$=!S8;YhLH}1XiE4s6l23t
zk&0|Z`Hg1BP_3RhZl<#=&C%RXA?{sBI1rc*i67ko&=gBkFl@mOMJ*TcX@K>15mO7#
zOlP==>3pVLD&n{BtcZ1Yp!S=3cjye}YWR6g_;wAdxE<;}3079Yu$$AP2ycdPWlP7f
zPdy4_Jx|6ZlyC3FSUPBRq}=Vdj6Oj)=H7&aUn0WWNNj2UN)W5~unNfZ?5P_@0G@@j
ze|LY@wzWgkNApdH`z|E>!3f_o7|yCF{|<}CsX$irhnd>=n;AIHQ}WM^8LV&_=t#Qc
z<N9ht_<fz=S`lUyuBatae8kl|hOM&eK*eo(f#<$vjx@kra0$+5n>+sD69#Uc?}nSS
z9Rt<pH!1>frsL>nEc)`J+$!OtuwJ+e)(UsWG~o`}!F?U!Xo2B=j69^M=nfz0sZL9E
z`M>Gwgz2=5F+3yI(01I!)A$!(?!*)RvPo>ll9nv56OL*d;GUR!N)xd2!ZfT_^F2mC
zZFAuD$X3%C>}T*MXK44l+iKews4aD%LK<giS+95Ev$V6v?{?Nj+kZ?-+R-=ELAMV$
zt`nB2_0_(IOYeDVzuz5U09QueHM<Vhrk?C3casS~n+Q_R8PmJz-Mz$Xir++)%Bk|y
zzpwlo<})Jg975<EzF3|iltEG(6%cBpT$|`5X~90i%KVM}*$B*d4ML(2yNu8XfdPJT
zkX#<yup>Is;YWHR+t#pAtY&KI@|AW}9ca#pd1^{p?fIxJ+kwu?_B6X<8O7>EzU;P@
zRMM4mrAsY6PHBqh>-JnzvkQa9b9o<Z7izKE(T=2^Q8@5UpUc?aR+XyTC}f9qC1v3_
zvX$*Nl!B`ltxRnqi?Rth9jODkI(jObG-{IX2N+w2^(3*|VoIw?Vr{P?>ctm;6`86c
zCd-%S=5Y?0no>>Hl}e{o%;P$SEoJOpKrMz>n&oxM$$G7HQBP-6n&a<l*p{UgD;IM$
zOj^Z)hx^ik`2QQWs^3f9t^($)hx-CpT~#T0fKeoUawHMvp0=lx8e4YN)6>?KX5wDU
z<Fc0-9bY07*E!d}bfZ?9s7iK7VqD6@6EicBoh0TJoPW?~poxU}a*9cjtK*RjhlE6g
zPn9zuD5lf7Oq50JbVJpEPkC{q)_kJAx&r?s_K`-I+IWIGc#U4*f$$Gc5IV#|rD%Dm
zd<>phyhQd=J|Hg$Nlj)1e_p)Ofs?wQ5+Jp4TmlVWBy=)?cn2`a1L;Ae>3zWqEM53Y
z3ug@V9|U*<q>vPv1MwiP?t_IS{w0I$K|e}3+#J3fPE}asHpU0VLMb|q7YJN$5Q>lY
n)c&iXF?g)^dXLG&5)U4Zxc{U?U&6oQ`49EiJ&HPmKhOUF`kGWX

literal 0
HcmV?d00001

diff --git a/tests/data/square_map/dataset/Norder=1/Dir=0/Npix=28.parquet b/tests/data/square_map/dataset/Norder=1/Dir=0/Npix=28.parquet
new file mode 100644
index 0000000000000000000000000000000000000000..61faf754480ca7197639c323b3a09c7e2f174048
GIT binary patch
literal 2881
zcmcgu&u`*J6gCcG*(8T(yS6~$fRJMCA)5lh*|4XM2@tahNr)B$-3r;pfFs+mHUvzx
z(w?dwtDbt=a}PcA*yE~)s^=c6{uMp8*S;B>pJ7#1P4@|gnfKnj_q{jYjJQCR6cc22
zn8&9K%}|dhDjcIci7^!Oh++3Yzr%!>P%ILjTIMsBda_0feh=d`jC3GCQ`|3s7!!>0
zRA?s}*=lw*+34%zW;)BzEY1EHVBZJ0BQE@J;z#!YG{w*q3_I{cQL9Cye5`kim|7y?
z=?oV!ozJx^Mf@6^7qRIKWV@;MhSp%AhMzXLZ#JNcyP-aiU~Ls)W_x-R!R-L9Y~>gZ
zsV70KkC1T%<+}$lhW1+>D);P`W)p;C_HBUsImEpS#a8yO1+k6~>wrwpo|-WN@GPDE
z`-k(kZ5-<!ny&-ww*l_=F!x;;!&w*QpJ7dTx~4xf4o7CQ6<B|4&0z)0Ku6Lg57(Dl
z+;5u%*EGdKn02_K22Vvoj@s2snb`y??$S#<_cRNn0p@~>e?GhH>Gz+6;pTa6xJkP)
zP<=L25rA_YOGjhTS0Ck8aUX^C+<mZ?yFaFJ_s9<Rs}M_bVfKg6V~UFIMM8bqYROLI
zFZwp2TP<x2&xkQJEhq6j{`nVs@r1W*5}UT7CCl4{C7T+!Cl;R41ng|<n$c>0!04rI
z4%|N3s#}A@4Bli7&AxkAZE6F#B@bjsV+{@C&0c(-cK-O?&AM#+k4Z^8Y)u~YY`}4w
zFm%P1Z5Nl`cV*k|jWB>4&35#jg|(??`^o)e0?;P>)C<P+etQ2P@rH_Qqe|smdhXrV
z$lLIzL^?Ty&?S7ayg(>}gbpeo)Iqrp(FwwmeT0?ynf=8G%y<n#q7b`?&<KG6esPdo
z9^0@ZI#J<AcqW?0uu-gLYU%Qga#9^A)|q~;3rg*3)DcZzXJq?|SuwO?bs}E(I#Md>
zh`G|Wk{+iNiT88|j;@%6LE}n11Y27!R=diH(ANq_?)jC7{Y_b_I*mehSeH@;jw2e`
zUPCH4YSGBlCL%BDkkb;nkgKbvvPrEbcz%GfbzDypyUC}Nn!wi%D!f{J30RRXOMJ3?
zeW{=3kgiD8WL>IsTg5!CW7v|$P8({`+|oR+RZiAxrFK1?O(~YQuVEU7QmnKWYM8W&
z1sC_F1@ZqkY*nw9s#yihSr_*OusX6-aseYx`eX?_%w6R`B{jCps;j1r8%4*x7RO~b
zGdjISI<9k}f9Xc8G?At3ki@tahi7_bB3enzD>#3@&p;Ca^W_|qB3H*F8IB2w2%jn!
zKu}DlOOYsx*l9*ppHF#yB-cEmzP<tfB=!+TnA&)Px_FIV;(_oFPY^oBL#1fAsC){Z
zS-eE%N<JVj2uV$51b?2tQGt`Hml7a#a9jcnUnF!kfp|wS$ph&@gz0_33oJeO3QK1U
z^&k0oeWZ{Sngj75uIhn>B;F;1?S4N>INSoh98OhO<~GI$#X>16j^_(pZxAXH@5}b<
kp*DD`_WMuC!xHx&j=1-vL|?&QfyEE?=L3qmfIrWF1J8L?FaQ7m

literal 0
HcmV?d00001

diff --git a/tests/data/square_map/dataset/Norder=1/Dir=0/Npix=32.parquet b/tests/data/square_map/dataset/Norder=1/Dir=0/Npix=32.parquet
new file mode 100644
index 0000000000000000000000000000000000000000..fc36e5eabf631d32618a0768de4d5288a2b83302
GIT binary patch
literal 2881
zcmcgu&u`*J6doMJvPllnc5Q*g0U^cOLpBA1vtdse6Ch?2k`OHhx)rjG1CDIN@*`lP
zmG)HiSoPG?o_pw_#~xQbR6X}l^{?o$z4p!61c$I%RZaH^#xw7|dGCAA-;8{UD=9X}
zZm^G!ScXN95DLeTD{&UFk63OC^c!r54aFkinPooX(32%v@Ov0%U}ORT2JycJVr(!f
zpwLD%vexYAirF(J&2*M!IEMQvz`YOfdwlrc#E)(PXv8uIh7EWiv{=MG=!o1cVxQ^}
ziiBq~Qp9XN*MghBh~I+qB31`OrQg)LBYWtp;pY|p+ZCwdZm16=SXxDxU7x**;CcW+
zELO0Ko&<?LLdOM^?;gZh#%pz`-0ipYK1Dd;-Uj$zLj1c>Y+?UW5X<<m49M)=Lo)%$
zO#E$o|8U;6l>@^?^G$&JF2Mg0=D!aUI5&HJQ9ceU?xkDOpYY5UxtW2pAJ(61b6CML
z(2;h@#r4%1|NAP%wIs|kTv1a%k<dWv=$68+0u?uzTRe9)KGFbl!Nq$&o7~Y4pM~M%
zxo)IMn=w%B->4|SxsGF^vFNK$a;o@G!g~HbSj*oZ)A)O22lsV|WB4%lW9TtL(XB|R
zr`RoJ5c!+AP8fDepTIR@jx2kScpm@a%dL3AT{fvrU(nL!b;4FG9o!SXt26;SFAUvm
zH9uf<(>8}rk8U;W;ckX(vPV|WxvRGHq0&-@3Z$_|ruk+oK2JNp{O)GGZTpW&X*>G5
zGVJyN$92Lq)V|VpNa;OC>G!*10^myT4~(u&w5ey?$?aqU&?dapK4W@2y}grogCgs=
zQaO>IyXQ6XHvBo2b`E2F27e+iFwS7HjSCpJajs2uqIk<bCd&Mc{lyr}WDP>2P`iZj
z7=r<xBuFk#Y{U^CYVZ)BNtQWk6swt9x_qS`R)?y6Y@8UPT00-NCCk&9*`8`uOubm0
zN|)WXoJtO)T<KCxPg1HZxVoKzp<0Du<6PPW+l5lBcGN?$rx*5|lXHpqTZ&vAGz!^K
zT~3)Kj$~%L4Y@GTie{!Zl?2IvoVM73TpcZyP3kq#^#hEpgL;zMEg_}WM4`4*5wzk<
zz={n;7Lw)5Gvg?S4OOlt>vE;jD&|QYqn11wTtF?lQ<~?s%gK7JbWu-dQ>yLmYgneK
z7AqIN8m6sc!6AKVLHz#>Th;BQW>o=m)**cXtOG?ZIe<~1eX>OX=8n3f(HdJ;)zQ-C
zm1>Y)OOvvb86RC@gVgEkU%F8%O%*viqA@O|(Xo-4N_LX)3f{lhXP}8l_;Nx>k*kxD
z4hNJ(luwmYASj{JnM9RE;&ek*k574FtkhhhzP<wgB=Hf)gxX|+J7kStl7aG%OfWtm
zL#1dsxO@bjS+c~|LOvib2u)3A6n|d0(twkin-U<kNn8qz93*@^g?M`~$ph&@#MyZv
z3nE?ki?`k})W7H9^^ihRXb!}KxS9(VlDLNqwtM|3k#IhIIg+Yyo7)5*6bq$jB%UX5
ryg^VT-c$OoNBZ!o*6Tf`7fal`IO6V=5`6_<#r_}q>j6Ti@bUZ)kw;cT

literal 0
HcmV?d00001

diff --git a/tests/data/square_map/dataset/Norder=1/Dir=0/Npix=36.parquet b/tests/data/square_map/dataset/Norder=1/Dir=0/Npix=36.parquet
new file mode 100644
index 0000000000000000000000000000000000000000..c8c5cba2b986f2c30fa3a7b0af38daa4f09cf7ce
GIT binary patch
literal 2881
zcmcgu&2Jh<6kpbeY-}H_G`j+c148jC2ge1>Iwq&iV!%smY!ihAnhNcgVXfVTuwYoM
zNI6wKRz3BUa}PcA*kjZ~)pHM3|B4>lYv0T+=EFo)RX0yC%)IyJz2AHDn-OQInr6f7
z7W?p+rCI6$MKN*8mq#qcK47^W&~LF3HWH6AQ_F0|QIA(>!S8;8hLH}1Xo~+O6lcRR
zfr@O!q8qJ_p;$e0+{)xwnxnZNL)`lif50>UCVp%OKvOJD!LS8C6t!H$d4TnH5mQT)
zna*$#)A>xhRK&00SrO~*K<T%%?$8;`)$r3A|IHdyaXZup60EF(VK=8o5#9{p%9f5{
zpL!g|`Y0KfP`<ktXX&8TkxI9(8-0Rs%)JfqKS%g?k@(X7l^|B}VHJ?+*;6-+06Yt4
z|Nj20ZEJ_7kLK$T_ic#(o#DS@Fq~CU{u$OIoC;(`f0(H)dNTvZc}o7VF@qJZ03At}
zd|Y2{@V~7STr0w?!WFXwDjIRMj$te8I#6+&Uf{W}nIjD_7hHn#+2)SF|Ac{?=eyx1
zZO1|N*^P<-oas0^7LUF9D7T9LD6Hr2g0=kJF^#`Nc5q)sIGSg;A0iJaDz+1i^b|)|
z+~{BQb<%WnV+_xTHMAW!`7H7I7dwfhzibhkv7{x->!hRD2Dm5Zp3)@jyf6(*Z+*b%
zr)>?q9@%O-gZ(Vt<P7becUx^614UN`3Z!v{mi1;QF-tpp{BCDmwEf4Vq#b=j8Fc%A
z<2q@XYG3Jlxb&W<^!wcr25@EcU9;<8ZTjhMYB!Yxw8<d#oH4VT+1*RNp`x3pT0NDY
z`S&&Ymid%OJC6`LhcA}r2xXDjMn#0$DBmVJQCzT(u(EJte?9^;UW1S*#4aH;LSTSj
z93)@BHtdLwH24vpO13p@mTK90rgEho)ds3_VxF3!T7NNWOLm~May`|qT1KfhkuJM!
zIh}H)eECw%jMJ(t__{sURPExR`9j(U+l5l9b<`uVXA}>-(-#u<w-veOHjBAoLrz;b
zj%4M!O}Xf5B`aH>NP=WSPDkuOu8x+@rHs1h`vFG%u#qBmTS%*QQK;`#1+DZFup(2D
zg;eG8+&s=BQ<ZC}hFtCFr2?*FsLNyb0%|e5@+_}YNj2)_i$*4wRvmv|)3z+NRK1w1
zVWO9c9`1_{@&7k$HNTgdT?5QH5BCMIx{6%(0HZ+q<cI>yJ#|kbHR^WF(=yhTYT{l?
z<BFFZ9bY07*E!d}OtW5|C~|H{Vq8kY6Ei!JoD}92oPW?~poxh2a*9cjZ{U#(hlE6g
zPt`LZD5lf7M3g1$bVJpEPX%G5)P179z5@Rg_7O*z+IWIGc#U4-f$$Gc5IV#|wPbmy
zatxk1yhQd=J|Hg$Nlj)1e?hp?fRmb^5+Jp4TmlVWBy=)?cn2^k0O>)*>3zWqEM54D
z3ug@V9|U*<q>vPv1MwiP=7WVK{w0I$K|jhk+#J3<PE}mwHpU0VLMa-K7YJN$5GtDJ
nDgD<&WAH@l^`4N2B@sLv3I9ony@G$m^B?NZdlYpBf1dvaKm=EC

literal 0
HcmV?d00001

diff --git a/tests/data/square_map/dataset/Norder=1/Dir=0/Npix=4.parquet b/tests/data/square_map/dataset/Norder=1/Dir=0/Npix=4.parquet
new file mode 100644
index 0000000000000000000000000000000000000000..9ece3d2b52173dc69bca7d0f7d98f814955866e8
GIT binary patch
literal 2881
zcmcgu&u`*J6gCcGy-5zyc5Q*g0U^cOLpBA1vtdsi6Ch?2k`OHhx)rjG1CDIN+7K|&
zN_(n$ta|Ec&pq_eV~?vIs-AnO`d9SWUi)Tj!jG`3s;2t{!_0ec-uvF0Z$_M<YMKqR
zTkPXwmS(9(6vf0TU&1WKK4Q5Y&~LF3HWH6AQ_F0|QBPKA!S7*$hLH}1Xo~+O6lcRR
zfr@O!q8qKQu9$se+{)xwnxnZNL)`lif50>UCVp%OKvOJD!LS8C6t!GLD8PERh^ZyY
zOlP==>3pVLD&p7htcZ1Ypg1k9H?#+HHT<;3f3pTv+zs`C1S_jx*v;utgg0@h<qGzx
zCt<9Ql5q*;y9aTW4q6?l^qjWt5QJmyZHWIl!oQ2em-ep&v5F6?fK1Pxx@835SvdRm
z4`*##J2ZSWUx&DFL;UXy{~d$jtcvo_u);VM$cp|jQ(N?Q29EQT{9|JVD_j9Ok}mnU
zzTDt{TPL_ygjt0vW(rg^;%Z&pQrLB%;x@g&b6+z@8elHC1n0BO9e@7`12@li!%f<b
zgX*(e6#+QYada#md-YLn75`CK&))}Y`TJuUe~;|ozKU=(&u~9P9#d3oCmQK1c3W|y
zf6+Hd!*1(ict*^jWxL7eiO;{-NhJMci`eufEm__qZN<{TJvsN3CSm8Lp_}d22aJB&
z*1+qNt%g0=&*Dw?(CT}4)s{X`+R8wIH1^Om-|Qr2X=jh$-K>kY|Cp4t!_k#N&jB1a
zNz+gr#qn_IeNS<m-UtJ@)*aXA*;t!?wwu~bB>`<RNIhrF>}GcNl5eQ!CaP9X<>&r=
zjlN|*CDO?wgwElM<pn}nBz8~{p$^J-h)xt2>?5o!+}dA^z>L=*Bnq)h2#pXJ;1>tU
z7qAUGq9YA{#AlLa4x6P~ww|e6t4FngYM&UVhN#vrMjgotbY`xvT2)go)h5zauOp{Z
zu9PodshM$Fl?7k7=NhV295gSaeXw0BrCL`#68n1bz&pK=u)n3qHMd#J4I6UW#Bn4u
z*K5i}S1Xy>`a}{W19IA87jkvAbS|aWMc)rFwhtRAVz-2}S{H@-URBUaF99nu6j?}B
zuFj3)JTg?dmTJh=Zo5>#bqw3`*u8{Wbgw+iYgbZ@dik=E$)#1>-`BKEQ!Q05=W3X=
zOGOX&r48}_H*7V(mzq@r%sCJD1+cn`T=oE?K>B2h0?a*iPa`$9t(vE0%xl%ay_Uum
zFFQKELI$pLu78<ky*yFm+>pe$l7=Tnb|Tp+%quwmpwB=P5%c8~lOo^1BN+|}i3p#n
zXFyO)r*nxYOW5g_ssW!0!bqw6M16e?{weGujxe?H1a<Kmy~G3IAD$p|h=*#)^ibs(
zJac%7tfhQFUJ#O+%n1I1aIFC+H9sXl>fpEp8oo&AWCHOHU{V0mgNW1nf)`kN@D&%%
z80tR=@CHaBDKrP-L0ruT3rYM-2HS&vlySH@e0iLzxX5jc4~m6SG#oDwxZWUCG|^X_
l*F$~qRO|Pjl7}S`JRAxCNr}CJf5r13>dyxhbq0T){|2-`P~ZRn

literal 0
HcmV?d00001

diff --git a/tests/data/square_map/dataset/Norder=1/Dir=0/Npix=40.parquet b/tests/data/square_map/dataset/Norder=1/Dir=0/Npix=40.parquet
new file mode 100644
index 0000000000000000000000000000000000000000..6f8c8f778a0f20866055314b52032a2c9753d359
GIT binary patch
literal 2881
zcmcgu&2QpH6gOVPvPllnc5Q*g0U^c8A)5lh*|4XM2@tahNr)B$-3s|D;K(+t4FMCa
zw5O`as;8dz+(QpN_PFYy>bZxie?^b&wQt5Ie1uh1HQgr|X5M@A-tWEn&4@Ep%P?VP
zlX-H?&<ypMqM`}Pmu-e(9y9C?=r@@N6G_CPQ_F0|Qrj!E;P)s=!$^lhG{yZAN-*I#
zPenH4vGsOWm(9L0ZfElh&C=|TA@+TUJK&=KCVqSeKvN7&!LSKG6t!H07GS+w#MBat
zPG`7?>3pVLD&p7htcW#tAUkcfH?#+HHT?8|`{n^uaW~Wl60EEu%4|%JBD@j8l`S2^
zKD8aj`WP9PP`-PZVCbOLk!sJ;bcY}uvu{J(&k^ojB(b!AC5TmgSOsKy_S7vS0MEkN
zzkf7q+k->HNAq=v{Wiq?9_7A^VmPa!{4=a=oC;(`f1*=c>~;o@^OXE!eFiIB1v-*0
z`MADZ=YCrwxK@N&g)46IR4n4EUEPwIHK5`ay})x{Ge;U=F1Q5ev&9~N|49^Xp6`a6
zw3PtWXSXT>aHeDFcq0DlqueU)qp+U457u(`$29I9*};AlVQDVP{t$UWQSqHvq%Yf=
z?8g40Z&HS>>0@|C%%NqwsprYhzt~Bp{AHWi^d&7>-lS~V(!o77_mrk!=cS>WTKfY=
zKW%&9^~qMl9_;7vCVOc0y}N2lAIO?KkRgpdG|e|V$ywUj<99dfqU}E>CGBu@dC+qJ
z$4$yK6i0SETzcP=9j7<K0Iqe%HF`GIW}fY)chf0An+j6T8MC|D-M!QsDz<@YwNvT2
ze_vy7qn{G#6c9q^@Wt{1p&SxAsDw}l6*@#G2n+TRRu*sVFGgU-YY-BJ*hPd!2n_Ix
zgA|I`h8@w73O~X#(K3gvay{3`R<D(#`arQyj8j8U8W*FEXazbm-&d@fsh8^$@v7I6
zGHF*VRIZflIHO3suiJAC#VQS27vetHF6DB)s~ib^y>#H6UWnM=lBK%aD&>bwDP!U|
zqM7ftq>`(a&0J$5@}dDbZJ`Ufx@sn$)*FKF2N<=(W}4V7KBF`QzOh&1)$&WgiVRue
z)77hU<G6qfMXINpQmw0%i@1)VCXL-os73cGv%GdS-E34Yo7sFuvHg85%QTg8?Q*V$
ziB>LoxGx&S|KG6H{a&h89WduT+!w&=%2LGxj6CU+E$}e+ls%Qys9AMS&6?MWfqN~E
zt6pw&e1!~L=Uo4?twv=cOZg#*aU~8<jNC-D)0kIq{z0FCCIaTmDJDgsiAORV5)u(U
z)y{yRm`>*+QI@gOEmZ?P75S0e@QM2R8vN7PM;Kvh;|c2GHF}8$!aqDg=nxOJvgx7f
zF?i<j5?M?6fV?0iHJK6oMgCd^PO5%NfYiZp2{e3>(8&bi9l)drqz4hE_XRJo^x!Kj
zoH5jY5a11vLQ-fB#Dlo14;GU6mkhQC{ixt@bNC84RcVpi7#|c1rKmVwAaK1ws93Tu
mJFkcO;Hld0KP3-KGI%(W{*w}a1^<fYKh&QODe4UVJpT>bp;)&7

literal 0
HcmV?d00001

diff --git a/tests/data/square_map/dataset/Norder=1/Dir=0/Npix=44.parquet b/tests/data/square_map/dataset/Norder=1/Dir=0/Npix=44.parquet
new file mode 100644
index 0000000000000000000000000000000000000000..3c64d2feda18b4a5045ebcffc9988c6f74a3de30
GIT binary patch
literal 2881
zcmcgu&u`*J6gCcG*(8T(yS9wP0U^cO1DgWD*|4XM2@tahNr)B$-3r;pfFs+m7y>3*
zX-`#;RZl(bxrZKl>~YmY)pHM3|B4>lYu}7b_z_lB)pVa=n0fEbd*6HW&4_bUNijiY
zhk1O$&<yp6qQX(ilLSLCj~F%q`W+_3grZz{W|_}e>d6`{_&to#Fw%hlP4T}3qD(L%
zP@$a&x7F<GirF_N&2*NbS(^PZz`hUghkW?o#E&EZG{w*q3_I{cQL9C`KGwTM%q(1Z
zHp4~C=5y^z5x)lKMQl1lWzf`mBYU_|!%rLhHycpJ-B2G$u(paYvpqYC;C294wsH&y
z)RQ3Ab7WjW`R+lKq5W2e%Dq8L9}t8S_HBUwImEvUMOXH(1+k6~>wwJ8p1NfO;8{BR
z_Ydc7+c+{jG+zhUZv*`AVg9=?hO;ipKf_AkR3K~m6Q0?)+Zi}6Qu2?jIjmqA=t#Qc
z;reom|80}tS`%g+u81j6T*%S7x}`9iK*e2piRYeXfi%EeaPiM)mp%FZlQ7&o&kZ+e
zHwvoHZdC-}T*uOpXynyLxmEl}VLg8ztmW^IY5YC1gZ(PR(tMcxA@rD{A_*?kSL~MJ
zaDUM^al>xu6L?0<k!3sa=dsVfNW|javPo?Eik2*I;<jSx;2vLiO5?DzZRlpJ`2nMs
zwmEeBWUFBh4>EX@J+k`lUA3hTm6kG8AdNjT%{PhIJnj7PyPI{{_8*gyb_{f7*c$+j
zo49GH17+ai()+G581%*%z_mVbjGm3Psb_o1y<{BF#{JX_#`Io#Z$JKq;<izxawb3b
z?ko2;{3(%64k2^_Uo0;W${?|W3J7&ju0wR9xMUw=W&YOwVhm=y1|d<1T|#J#zyQBE
zNG^|U*byCT@FPBxEOXQ-Rx`D9`C2`$4psZqI5R}Gb~)}ymaj9jebuU%da*i{u6i9g
zm2{+B=}JvcQmQO?x_!q`t-`QzDII{VtrV+W^;qodg+ur3Qo{b0B3GS8Av>zeDHF$$
z%xteA7aXl<W@=MOkPOIai(Sao)l%7{UK2e(z}PyfCyCt>Qff^UYWo#IE4~D*$WUY<
zS-!e3PIAal<!Z7nSGuiY9@jBy$rGmywdihXp4ToX>$Ot5p3bIJ+uPT$Oj9jZ+6y&I
zTg8Hl`_h8={~NZd*GtW+0_LoX`vO=UMJ~C3Q6PP?MFHlny04KMTUOQ8(&n{l;9g6U
zvYQ#7Tp<J3xzN9KqgI+Ka&|;wTuGx-BQurkB<2;Izu#w|iHP}fhDnjD<B<$UghYf-
zm2)5{rqhK)ltt`xOI4pwd10*7Jfgn72LB}X5yzO?c!Iikjb7q`@DEQAI>JMxXu7C;
z0-jmCMAk|^ATJ0>O=bjtUbxnPlbV+jAa!tD0u5gzbUKB2hcL+l=|RNVeZdPXJ@|@C
zXAJcp`gnb$kQAB&@gT0|frTXAC4=pLKT0^<0=^thRaoXW#s|ehDH@LF3tVpyii`D?
m!RwJee5&>PPszg)^B<0w_oPH#!N20g5B28*iaLis&wm4CJX#n4

literal 0
HcmV?d00001

diff --git a/tests/data/square_map/dataset/Norder=1/Dir=0/Npix=8.parquet b/tests/data/square_map/dataset/Norder=1/Dir=0/Npix=8.parquet
new file mode 100644
index 0000000000000000000000000000000000000000..214219102fb9915c793bd02d41d529d1b0f6fecb
GIT binary patch
literal 2881
zcmcgu&2QpH6gLiHy-5zyc5NAn144?mhinQ2XTzR4CP2(4Bq3T1bSvbqfFs+md<0Ci
z(w?dwtDbt=a}PcA*yE~)s^=c6{uMp8*S;B>@DWy3)pVa=n0fEbd%ySQHzUqaCB=r>
zE%xy-OS9A?ieh4vFAPhuk63O8^jmC%jl_6nYMISA>d6W%_&to%Fw&tAO$on*Vr)1n
zQjx7FztL=)s@=85&2*NfIhy-1#Jvv*2LkhN;zxG?G{w>s3|sI+QOiYy1FUz8m|A#d
zI>SXw=QHh65x<6KMXdJ+YOkqxhVEdlhM(4iZ`PoSyP-aiU}Y5yyE#3I@MZ{CwsZ{p
z)RQpQ^JH8?`R+lCrGr*S%AKBJ_6Wi;_ckQ_91-3{VoUp1f>_0eRY0a^Pu(&C@GPAD
z`-ii(tsPoEny*9Lw;|zoM);1wa8^b6XIRr+EBeDsZT#(Q3DzGQGg#p=(2;b>$Mxlg
z@Y_1UwIa+aTv1!3_()%Gn~ut^0~NRF1)lqwInn@g!6i7KZSMH{PZ+p)z8h}Rb_`UX
z-Kq${nU15QvFNLha;t=o!g}F8SS#Eg(}a6u2lrKkqXmZhA@Z1_qC0$~tGb5T=l`N_
z5|(S2V|YgFq2u-w&*Pteu@g`D%O<gzOIotLNw}(Gf_q}_DNVr6OUty4<_C;^+UCIP
zlC73I*w5fi?$GIach!zLPz`mULK=5y+i!N_v$V6v?{3yb+kZ?-+R-!BL8k{eZW6Yo
z_0*n+OYeGWuh$u20M}-(Z*^R(O+DL9?j{p}HW8$rGp2XbyL*W@6u*fol~d)pe_#2x
z%%?<JIfT$Te6hShD1)RHDj?KCxfan$(t>@2mHAuyixHUd8iYh4b{U}&0t5WwAh|rY
zVMlbN!;kb#cI;uJSk2VZ<!kMzI?&t`>(r98+Qq0PJAuy5b~UGBo5kuxzUs7;RI)GU
zN>^HXoYEB0*X{K!%_$5T7xF&XF4bbStsO~SvvA;@UdY(rQI%@HQOFMKO3KD@WINkw
zD22XWv@^AdEXo$-bfq@rYU`<N(yU3oA7C^N>q%mF#FSQ(#M)j()Qc|xE3#BYOqQ?C
zt>YZBG^LuXE0wlU%;P$ShBEG7LM^6On&oxN$$G7HSx;wEn(OatIJT`7E0=RMOpIc|
z!+kLz{{M!p>i1H2s(?A`;l2RYzN(Zwz$lVFxsnKTPutT;jfPY8^t64gS-98oxa?&{
z$5+V0b<XuK-Kdo&s*)X&7+3P}#L7%$H;H)#=O6SLXd+?0oMKYs>Ubo>At4drQ{@Z@
zis^JN6J-%Q-BLB+Q(hdYHJ_-jufacweWVelHlCn1UZa<IApFA<gbwjgDcT+?AA@HW
zFOjp956BBbQj;0MpBJxn;H2)S1V}9$mq5c837t$J-T_SVKza~qdSCDYO9#Hv!Wl#T
z2LavyDI|sFKs<=6`(Po7f5~8b(2o)hH-|5WQxz7ujqyRTP>PP@1p?O_gyQ2}wfA~x
i4xZ}W?o;xx#Dj+;?msEfSMaZR{zLuwfTGUe&-35tbWv9T

literal 0
HcmV?d00001

diff --git a/tests/data/square_map/dataset/_common_metadata b/tests/data/square_map/dataset/_common_metadata
new file mode 100644
index 0000000000000000000000000000000000000000..71be47909c9a7b5480c2882595038cef733d0c20
GIT binary patch
literal 1612
zcmcIlOK;mo5GGv12-2KdIFNxH1OmP}E+vbJnNyasL{qV4n?^`VZ6Pcll2%w!3?Gsx
zBS3#p|6i}2S+cgApr@h(ak)D)JM+!#x701MT8H~bT$LIJnLiIBdiYX0+&|uH{5YA1
zlqYO6X=|0{WtP));xCIhmk#&9<zTo>gD^dQef8(hFMg7w%I${>mL9(W)yGHG{t)6k
z>0V1muZ~U{2MLV>nl=8~dyrWigd2#>izv?I56xeHKWWO<aw2TuH(IfLkYgH!;4bfg
zm0{<Ug*=%2i?K>O$;(7+WpVzlgPY<aO3LrmQJB+!<`mMz1?OK+nont;I`Vzi=eGZt
zRJ0=vX}(MW$Aiq7m(sMv(vy;=>2i$#R$-d4WsKU^`%~pqkpZn-r`|EPPur(w@>i+x
zidfc-^PxKV#vd<!5i;u%Lhj%}`9w&E=rhtMWJbC(q0{we_BASdkM>V%Fyk6TL=kqA
zkTn4VyfH|(hc<L1mp;7o_h!V4v0-=I_HgB0+PN2Bvm2&+?)`dZMzxM>i5FQsH0-Up
zSk9c5l9}DX!fS6@o}*T}vy6FBKOf(l@4z;thCTN#^(5?{mpAt&`bX5UvvFT5Mox=k
z9FuFyvD43dgLmAmshSLO#`+v`&Ha|9gsxus0mk5BqzHSYwmer?-7`z|jn9CUFzTqv
zaB;`3x`cU-t&ALN9vD5WqX?W$HicTka`2Ql9x5YuFdenEmKRt1#u4Y9VNG{x*ak+w
z#J&U&|9`_~SH1Kj8!&4n_64wJ)ESh3Q5AiPbrt5Ncjk*4gUBxZHeY!Rdu?upWoLc0
zAPno=>0f*74z|?M3K3&r7T2t^HDd*N1?OM)8EB#-Uv7{T-4TvrxDX@~e6ntVph%}X
zQz#8|dZcR2r=Gf|Zbj5DEAUs)M_(hgaYE*}CZBN-{KE+$7dTi3FUjx<JT+XB=$m{%
zUJ#L5%mn_Py7GaOew7j+%`mQjh8KxkZz0|}OnN|i5dEL~f(w)-Jo>XUhWgKIyfsot
z3eACd5ZAB3LXzr|!S=cz0}Qu=uZyYr&$*5Gpjary$9T2C^#+j|&4i|3iZFlcC&^p!
cVQJPMj%M{qIr{C5B<=pBj|Y-;3;!d30|Bbzm;e9(

literal 0
HcmV?d00001

diff --git a/tests/data/square_map/dataset/_metadata b/tests/data/square_map/dataset/_metadata
new file mode 100644
index 0000000000000000000000000000000000000000..fc7658086f6d5e36fc8e023fc33e68d05b21867e
GIT binary patch
literal 8773
zcmcJV&u`jh7{?7ojJ7#MTR}!*6GAP@fu%qo4V5a@kU&XO(ohvh)+w^V0Y|o>{0MQX
zN<B?GPCM<?^A0=gxMQ`$w8Kt2O#3T#-F44v^ZrQdC~v&lO3c^K`}%#p&*Kk{5qT;#
zbJ0*-FA8FK(jBOV9K9h#LwADW&BK8z>m%*-Fkf;aHzregeApe%%&8C!!jy1hIPR(A
zJF%UgK8t)R2+p)?o7(tJ0zwZ-DAa&;Xe=hi|A@a3KX*=lkh=?KJjmWflOAMONAmdP
zOr6Gl4<x`@Q3x-`&zXq^W5L)jf!Mo1dLtc4h!IF^Z&byE6_=<G&qPB}ur9!E8D0e8
zG7o=-s0XHe#=@`H(%)b6*TmaT64BtI5sQ}F22Nvd0_oqv>9^s;y08)opnB}RheF?7
zr-h;mXeanC5c?sJ{yUQXF@iFDoR<D`lV-BHU{ZZ^GyT^M{#3DV!?9pG68kxPOAsys
zs^A%@5CJO00L4VP`%FFX1BEUp+QbM}$OP6j>cND6K+!_|28w3l11jWn6WWQDOXIkJ
z+VBiiga8#`fMTL9mfY`9)B`_IxNh13Wa7i92NV7QMGN&CD4K~6D10{*BUrgK?tP%X
zaQl;&%X=nji4e8K5XD4YJc*Z?;795@c~fD9TwpB{<;BHEVd}C{KO$6@rQ)M%iF6bz
zm)`MFweFd!WkS_5LlqNs@pJ)I)C50O%cKWbAs1MSM0s)HBUQ9i->IUh_^9IQ9T!!3
z@6YGA2vxTjs+cJE^<+%&QxzvvVTD{^EfVF$g@05rrTR@3lZuZjc<8n_F@lv#@A#;q
zPjgrO)F%?EM20FR>f*`$UX`aN_^G;0dVm#jfwf4K7Z*NKMN9RaDw>LqDv@*)E0^B!
zQAHo=u1XdCK%Zc!VxlgdE})8<;HN4_dVm#jfwf4K7Z*NKMN9RaDw>Lqss!mMRxZ8c
zqH4|i(|VHJtC9>=Ow`4bJ5`JceySdl9$<xBU@a2m#f5)VF{S!V6_bjOswC+sRxZ8c
zql!M#UG-jdn^1L|p^AyRc)EZpYJ#7tJ<<cLkPECuqP)29kt$lM?^MxLd{l8gJ#bM~
z@cy*ELa16{sA8hr*OM{9PgRdlg%xsvwMdi~7yePjl<GHCOe#LAR!B#&a_Jo(RrHbW
zs`si4p(?{r#Y9~^x!<b_)C50O7U=<2$OYCSQC?j5NEI#BcdBSAKB_XLqgc80?mSf)
zF+7sZo;(r%4ZccgW=}nZFBSEfVNO!_lV5$ENv52`LwihpFKQpYN|~~u!t@mLEoBPM
zJknIXcX+{|llE|GjqFo3bGlwYCz&&2WPKPm)T!K)r!u55XS)76lcZ_sSD_zfJwNt8
zliKYVtMYU>1|P3dx~7cfv4u(>S@L*1oFfA#>Ug3JO%%=DTg|R!Q{XnWNX-o9SM#fD
zsn>$IBsH76?fcGGzT%t6U3=?SBuUzXU$i}zq=HoLOI1ngOO?JIE0@oW&rxuTj6a^k
zFgk|KUd0}-OVV6|0eD4=RJPC<nkYT$!b|yH-Oy(TwN{~%Z=5KPT2sY*qU~yBrSojw
zuN#Y)UK%MzQ&(%Pv-<I{-_B(x^~(0Kl0VHU?V=M~n`nwrogO@^ufy1pTx$)KN9B=P
z-LQ6_)zN%IZnq`})zWOgozu}eb-gq^XjdoQnqKIf)r)lva+>7<<QjByrL5X1JM+L}
zZ*xCuj~m6D(kT}^Yt3S}_8h!Qn%pjC8^?RvPDRp`b}PHzZVr02EmX&>*FK#bK`pAa
zP4k+K?0#qaXg^=dDW-GYfuZY4t$D=MaMr6;E!3ADtnZt*meb3w(E`sU3-txOPUQBs
z1s;obpUiR*w56<d?HYSV%j)L!6GcP4uAern!hGjg(omgD|MCZ&?K8Pun%Qd{*Jn?(
z!dczSBCg>2FZv9alo2nx2#U%++S|KL8zP&h=2HL^;j~w`gEcgXpt|5`t2mcC4yZ3r
zV15?ODbEqwNJs;8EImg%n?EEZX%p?5HQkaLJ210^4khEgd;l+Oc4}MM{<n%JU0|~7
zqy$TSw61Lpy^*9RXRzJ|h%JB~wlaRdpaZlG;kSJ58bkdX3%?7ekQAB&>%qER$1Egq
z-ehq4q95C6ImTZFrK+B18~KA`p_DFKZ?VDK8#X~qF8=T0%kF6OrF~;bE^Zu2=av$G
S0Uw>&AEA^Igs1QiB>w>f<C>oU

literal 0
HcmV?d00001

diff --git a/tests/data/square_map/partition_info.csv b/tests/data/square_map/partition_info.csv
new file mode 100644
index 00000000..798921b7
--- /dev/null
+++ b/tests/data/square_map/partition_info.csv
@@ -0,0 +1,13 @@
+Norder,Npix
+1,0
+1,4
+1,8
+1,12
+1,16
+1,20
+1,24
+1,28
+1,32
+1,36
+1,40
+1,44
diff --git a/tests/data/square_map/point_map.fits b/tests/data/square_map/point_map.fits
new file mode 100644
index 0000000000000000000000000000000000000000..3f1bd5f0b3c291a66c52132696bb535d8daaf968
GIT binary patch
literal 8640
zcmeI0!EVDK42HKi;g;(*RoX5+5aP6~1R|hP@&G;b(jD4Kef&AJyOBnLYMLr@LBisX
z{b3>-C!1CQV@nnpKC(8kR7&5<Zba{6Vspf-c_$sNNBw(=Y?xSr&V#N-^b+C?ViVsa
zGZ7#HM1Tko0U|&Ih`{3z7|KZ}jb5^%l-l!BL6H;PpVrTB^F0mq`0GghoLb+bAF=v=
zk)YQKv(;F~$Mjf2zp%QchaBb(UrSJxi~E<N=ld4Ic@)3hOIH5*+Sulzx4FiarH9Sn
z;339atbwbcH_XyAwgM~l3iUKN57^3O=?R01u6L*>K!u32^sv9wyb1B1PrJi$|8>aH
z!=Foq?Rh+367Fv;ts6S@d)*JkgIv_V#rxdb!I;!l=ft|f^B7(xq>w&~pD8~<g#STN
zZ<TVJJ~Lk|aBu98&jPP;$m#u=+3({i$xnJen4crGg9s3T#~|=@8PC^wY$E#@nfvg&
Xtxi@VKm>>Y5g-CYfCvzQe?#C0dh`8c

literal 0
HcmV?d00001

diff --git a/tests/data/square_map/properties b/tests/data/square_map/properties
new file mode 100644
index 00000000..c28a0585
--- /dev/null
+++ b/tests/data/square_map/properties
@@ -0,0 +1,12 @@
+#HATS catalog
+obs_collection=square_map
+dataproduct_type=map
+hats_nrows=12
+hats_max_rows=1000000
+hats_order=1
+moc_sky_fraction=0.25000
+hats_builder=hats-import v0.4.1
+hats_creation_date=2024-11-20T19\:32UTC
+hats_estsize=52
+hats_release_date=2024-09-18
+hats_version=v0.1
diff --git a/tests/hats/catalog/loaders/test_read_hats.py b/tests/hats/catalog/loaders/test_read_hats.py
index fd8511e3..ad37b683 100644
--- a/tests/hats/catalog/loaders/test_read_hats.py
+++ b/tests/hats/catalog/loaders/test_read_hats.py
@@ -8,6 +8,7 @@ def test_read_hats_branches(
     small_sky_source_object_index_dir,
     margin_catalog_path,
     small_sky_source_dir,
+    test_data_dir,
 ):
     read_hats(small_sky_dir)
     read_hats(small_sky_order1_dir)
@@ -15,3 +16,4 @@ def test_read_hats_branches(
     read_hats(small_sky_source_object_index_dir)
     read_hats(margin_catalog_path)
     read_hats(small_sky_source_dir)
+    read_hats(test_data_dir / "square_map")