We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 882b608 commit 52834cdCopy full SHA for 52834cd
1 file changed
sdks/python/apache_beam/runners/portability/fn_api_runner/fn_runner_test.py
@@ -454,6 +454,15 @@ def test_pardo_side_input_dependencies(self):
454
ExpectingSideInputsFn(f'Do{k}'),
455
*[beam.pvalue.AsList(inputs[s]) for s in range(1, k)]))
456
457
+ def test_flatmap_numpy_array(self):
458
+ with self.create_pipeline() as p:
459
+ pc = (
460
+ p
461
+ | beam.Create([np.array(range(10))])
462
+ | beam.FlatMap(lambda arr: arr))
463
+
464
+ assert_that(pc, equal_to([np.int64(i) for i in range(10)]))
465
466
@unittest.skip('https://github.com/apache/beam/issues/21228')
467
def test_pardo_side_input_sparse_dependencies(self):
468
with self.create_pipeline() as p:
0 commit comments