CSS Flexbox: the width calculation of flexbox with flex-direction: column and flex-wrap: wrap

The green boxes should all be within the black box