Skip to content

Commit 04a70ae

Browse files
committed
Add test for reading vector columns
1 parent 5fa0b83 commit 04a70ae

File tree

1 file changed

+13
-0
lines changed

1 file changed

+13
-0
lines changed

fitsio/tests/test_vector_datatypes.rs

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -110,3 +110,16 @@ make_test!(
110110
ColumnDataType::Double,
111111
3.1415926535879323
112112
);
113+
114+
// integration test using file generated by `filegen/vector_columns.py`
115+
#[test]
116+
fn read_vector_data_from_example_file() {
117+
let mut f = FitsFile::open("../testdata/vector_columns.fits").unwrap();
118+
let table = f.hdu("info").unwrap();
119+
120+
let string_data: Vec<String> = table.read_col(&mut f, "target").unwrap();
121+
assert_eq!(string_data, &["NGC1001", "NGC1002", "NGC1003"]);
122+
123+
let vector_data: Vec<i32> = table.read_col(&mut f, "index").unwrap();
124+
assert_eq!(vector_data, &[1, 2, 3, 4, 5, 6]);
125+
}

0 commit comments

Comments
 (0)