test: Add tests for save_for_aiu functionality w/ tiny models #316

Job	Run time
Build and check packages	22s
Publish release to pypi.org	0s
Publish packages to test.pypi.org	0s
	22s

Provide feedback