1
1
"""Test pixel path creation"""
2
2
3
+ import os
4
+
3
5
import pytest
4
6
5
7
from hats .io import paths
8
10
9
11
def test_pixel_directory ():
10
12
"""Simple case with sensical inputs"""
11
- expected = "/ foo/ dataset/ Norder=0/ Dir=0"
13
+ expected = os . path . join ( os . sep , " foo" , " dataset" , " Norder=0" , " Dir=0")
12
14
result = paths .pixel_directory ("/foo" , 0 , 5 )
13
- assert str (result ) == expected
15
+ assert str (result ) == str ( expected )
14
16
15
17
16
18
def test_pixel_directory_number ():
17
19
"""Simple case with sensical inputs"""
18
- expected = "/ foo/ dataset/ Norder=0/ Dir=0"
20
+ expected = os . path . join ( os . sep , " foo" , " dataset" , " Norder=0" , " Dir=0")
19
21
result = paths .pixel_directory ("/foo" , pixel_order = 0 , pixel_number = 5 , directory_number = 0 )
20
22
assert str (result ) == expected
21
23
@@ -37,7 +39,7 @@ def test_pixel_directory_nonint():
37
39
38
40
def test_pixel_catalog_file ():
39
41
"""Simple case with sensical inputs"""
40
- expected = "/ foo/ dataset/ Norder=0/ Dir=0/ Npix=5.parquet"
42
+ expected = os . path . join ( os . sep , " foo" , " dataset" , " Norder=0" , " Dir=0" , " Npix=5.parquet")
41
43
result = paths .pixel_catalog_file ("/foo" , HealpixPixel (0 , 5 ))
42
44
assert str (result ) == expected
43
45
@@ -58,7 +60,10 @@ def test_pixel_catalog_file_nonint():
58
60
59
61
60
62
def test_pixel_catalog_files ():
61
- expected = ["/foo/dataset/Norder=0/Dir=0/Npix=5.parquet" , "/foo/dataset/Norder=1/Dir=0/Npix=16.parquet" ]
63
+ expected = [
64
+ os .path .join (os .sep , "foo" , "dataset" , "Norder=0" , "Dir=0" , "Npix=5.parquet" ),
65
+ os .path .join (os .sep , "foo" , "dataset" , "Norder=1" , "Dir=0" , "Npix=16.parquet" ),
66
+ ]
62
67
result = paths .pixel_catalog_files ("/foo/" , [HealpixPixel (0 , 5 ), HealpixPixel (1 , 16 )])
63
68
assert expected == result
64
69
@@ -106,6 +111,9 @@ def test_get_healpix_from_path():
106
111
result = paths .get_healpix_from_path ("/foo/dataset/Norder=5/Dir=0/Npix=34.parquet" )
107
112
assert result == expected
108
113
114
+ result = paths .get_healpix_from_path ("C:\\ foo\\ dataset\\ Norder=5\\ Dir=0\\ Npix=34.parquet" )
115
+ assert result == expected
116
+
109
117
result = paths .get_healpix_from_path ("Norder=5/Dir=0/Npix=34.pq" )
110
118
assert result == expected
111
119
0 commit comments