diff --git a/rpg/enzyme.py b/rpg/enzyme.py index 9bfcb193cd8a5675247acd1977d920d7d75e1620..02260f9496a37daffdbb1b0e7ab225e804288487 100644 --- a/rpg/enzyme.py +++ b/rpg/enzyme.py @@ -78,9 +78,10 @@ class Enzyme: :param enz_file: location of user file (default: ~/rpg_user.py) :type enz_file: str """ + print(self) if self.rules != []: # Comment and first line of the Enzyme - ret = "\n\n\n# User defined enzyme " + self.name + "\nENZ = []\n\n" + ret = "\n\n\n# User-defined enzyme " + self.name + "\nENZ = []\n\n" # Write all the main rules and their su-rules for i in self.rules: ret += i.format_rule() diff --git a/tests/test_enzyme.py b/tests/test_enzyme.py index 90de9ac9ae70e1e166fc0650b3e74de2a8eaee03..64363bb775fbf7e26761c31a48010e197d593af7 100644 --- a/tests/test_enzyme.py +++ b/tests/test_enzyme.py @@ -49,7 +49,7 @@ def test_enzyme(tmpdir): all_rules = rule.create_rules(dict_rule) new_enz = enzyme.Enzyme(-1, "fake_enzyme", all_rules) new_enz.write_enzyme_in_user_file(str(output_file)) - assert output_file.read() == '\n\n\n# User defined enzyme fake_enzyme\nENZ = []\n'\ + assert output_file.read() == '\n\n\n# User-defined enzyme fake_enzyme\nENZ = []\n'\ '\nE_1 = rule.Rule(0, "E", False, 1) # Never cut '\ 'after E, except...\nD_E_1M1 = rule.Rule(-1, "D", '\ 'True, -1) # Always cut after E, preceded by D, '\