def test_import_styles(): red1 = SSAStyle() red2 = SSAStyle() green = SSAStyle() subs1 = SSAFile() subs2 = SSAFile() def prepare(): subs1.styles = {} subs2.styles = {} subs1.styles["green"] = green subs1.styles["red"] = red1 subs2.styles["red"] = red2 prepare() subs2.import_styles(subs1) assert subs2.styles["green"] is green assert subs2.styles["red"] is red1 prepare() subs2.import_styles(subs1, overwrite=False) assert subs2.styles["green"] is green assert subs2.styles["red"] is red2 with assert_raises(TypeError): subs2.import_styles({})