tests: Updated test code for change to Layer.__init__ params.