The context of datamodel is used to expand the RandomVariable when required. Optionally, the user might find easier to use sample_shape instead of using the context datamodel.
Multiple tests should be done to test this behaviour. Moreover, at least one change must be done in order to make this working correctly: if sample_shape is used, the Random Variable should change the attribute is_datamodel=True.