def test_initailize_InitMassFromTotalMass(): data_arrays = mock_append_data_arrays(mass_array, num_elems) substance = OilProps("oil_conservative") spill = Spill(Release(datetime.now())) spill.release.num_elements = 10 spill.set_mass(num_elems, units="g") fcn = InitMassFromTotalMass() fcn.initialize(num_elems, spill, data_arrays, substance) assert_dataarray_shape_size(mass_array, data_arrays, num_elems) assert np.all(1.0 == data_arrays["mass"])