Merge pull request #66 from astronomy-commons/issue/50/plot

Proper handling of ints for writing metadata.

Author: delucchi-cmu

src/hipscat/io/file_io/file_io.py

@@ -127,4 +127,4 @@ def write_fits_image(histogram: np.ndarray, map_file_pointer: FilePointer):
             value at each index corresponds to the number of objects found at the healpix pixel.
         file_pointer: location of file to be written
     """
-    hp.write_map(map_file_pointer, histogram, overwrite=True)
+    hp.write_map(map_file_pointer, histogram, overwrite=True, dtype=np.int64)

src/hipscat/io/write_metadata.py

@@ -113,7 +113,7 @@ def write_partition_info(catalog_parameters, destination_pixel_map: dict):
         "Npix",
         "num_objects",
     ]
-    data_frame["Dir"] = int(data_frame["Npix"] / 10_000) * 10_000
+    data_frame["Dir"] = [int(x / 10_000) * 10_000 for x in data_frame["Npix"]]
 
     # Reorder the columns to match full path, and force to integer types.
     data_frame = data_frame[

tests/hipscat/io/test_write_metadata.py

@@ -115,6 +115,23 @@ def test_write_partition_info(assert_text_file_matches, tmp_path, basic_catalog_
     metadata_filename = os.path.join(tmp_path, "small_sky", "partition_info.csv")
     assert_text_file_matches(expected_lines, metadata_filename)
 
+    expected_lines = [
+        "Norder,Dir,Npix,num_objects",
+        "1,0,44,42",
+        "1,0,45,29",
+        "1,0,46,42",
+        "1,0,47,18",
+    ]
+    pixel_map = {
+        tuple([1, 44, 42]): [44],
+        tuple([1, 45, 29]): [45],
+        tuple([1, 46, 42]): [46],
+        tuple([1, 47, 18]): [47],
+    }
+    io.write_partition_info(basic_catalog_info, pixel_map)
+    metadata_filename = os.path.join(tmp_path, "small_sky", "partition_info.csv")
+    assert_text_file_matches(expected_lines, metadata_filename)
+
 
 def test_write_partition_info_float(
     assert_text_file_matches, tmp_path, basic_catalog_info