add test for Series.toTypedArray null handling

pola-rs · Apr 24, 2024 · ed47f11 · ed47f11
1 parent 6b46542
commit ed47f11
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/__tests__/series.test.ts b/__tests__/series.test.ts
@@ -656,6 +656,12 @@ describe("series", () => {
     const actual = s.round({ decimals: 2 });
     expect(actual).toSeriesEqual(expected);
   });
+  test("toTypedArray handles nulls", () => {
+    const s = pl.Series('ints and nulls', [1, 2, 3, null, 5], pl.UInt8);
+    expect(() => s.toTypedArray()).toThrow();
+    expect(() => s.dropNulls().toTypedArray()).not.toThrow();
+    expect(s.dropNulls().toTypedArray()).toStrictEqual(new Uint8Array([1, 2, 3, 5]));
+  })
 });
 describe("comparators & math", () => {
   test("add/plus", () => {