Skip to content

Commit 52834cd

Browse files
committed
Add failing test
1 parent 882b608 commit 52834cd

1 file changed

Lines changed: 9 additions & 0 deletions

File tree

sdks/python/apache_beam/runners/portability/fn_api_runner/fn_runner_test.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -454,6 +454,15 @@ def test_pardo_side_input_dependencies(self):
454454
ExpectingSideInputsFn(f'Do{k}'),
455455
*[beam.pvalue.AsList(inputs[s]) for s in range(1, k)]))
456456

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+
457466
@unittest.skip('https://github.com/apache/beam/issues/21228')
458467
def test_pardo_side_input_sparse_dependencies(self):
459468
with self.create_pipeline() as p:

0 commit comments

Comments
 (0)