Add framework tests for tfl.real and tfl.imag

Change-Id: I665acac9b5171efd0c5a2b68b516609048f6e187
Signed-off-by: Luke Hutton <luke.hutton@arm.com>

1 files changed, 14 insertions, 0 deletions

diff --git a/verif/frameworks/test_builder.py b/verif/frameworks/test_builder.py
index 6302865..f872888 100644
--- a/verif/frameworks/test_builder.py
+++ b/verif/frameworks/test_builder.py
@@ -1225,3 +1225,17 @@ class TBuilder:
 
         def eval(self, a):
             return tf.signal.rfft2d(a, self.fft_length, name=self.result_name)
+
+    class Real:
+        def __init__(self, name):
+            self.result_name = name
+
+        def eval(self, a):
+            return tf.math.real(a, name=self.result_name)
+
+    class Imag:
+        def __init__(self, name):
+            self.result_name = name
+
+        def eval(self, a):
+            return tf.math.imag(a, name=self.result_name)