Bugfix: return 'f' as single register

Author: boriel

arch/zx48k/optimizer/helpers.py

@@ -254,7 +254,7 @@ def single_registers(op):
         op = [op]
 
     for x in op:
-        if is_8bit_oper_register(x) or x.lower() == 'sp':
+        if is_8bit_oper_register(x) or x.lower() in ('f', 'sp'):
             result.add(x)
         elif not is_16bit_oper_register(x):
             continue

tests/arch/zx48k/optimizer/test_helpers.py

@@ -89,4 +89,11 @@ def test_dict_intersection():
     """
     assert not helpers.dict_intersection({}, {'a': 1})
     assert helpers.dict_intersection({'a': 1}, {'c': 1, 1: 2, 'a': 1}) == {'a': 1}
-    assert not helpers.dict_intersection({'a': 1}, {'c': 1, 1: 2, 'a': 2})\
+    assert not helpers.dict_intersection({'a': 1}, {'c': 1, 1: 2, 'a': 2})
+
+
+def test_single_registers():
+    """ Flags also for f must be passed
+    """
+    assert helpers.single_registers('af') == ['a', 'f']
+    assert helpers.single_registers(['f', 'sp']) == ['f', 'sp']