MLIA-843 Add unit tests for module mlia.nn.rewrite

Note: The unit tests mostly call the main functions from the respective
modules only.

Change-Id: Ib2ce5c53d0c3eb222b8b8be42fba33ac8e007574
Signed-off-by: Benjamin Klimczak <benjamin.klimczak@arm.com>

1 files changed, 0 insertions, 30 deletions

diff --git a/src/mlia/nn/rewrite/core/graph_edit/diff.py b/src/mlia/nn/rewrite/core/graph_edit/diff.py
index b6c9616..0829f0a 100644
--- a/src/mlia/nn/rewrite/core/graph_edit/diff.py
+++ b/src/mlia/nn/rewrite/core/graph_edit/diff.py
@@ -13,36 +13,6 @@ from tensorflow.lite.python import interpreter as interpreter_wrapper
 from mlia.nn.rewrite.core.utils.numpy_tfrecord import NumpyTFReader, NumpyTFWriter
 
 
-def diff(file1, file2):
-    results = []
-
-    dataset1 = NumpyTFReader(file1)
-    dataset2 = NumpyTFReader(file2)
-
-    for i, (x1, x2) in enumerate(zip(dataset1, dataset2)):
-        assert x1.keys() == x2.keys(), (
-            "At input %d the files have different sets of tensors.\n%s: %s\n%s: %s\n"
-            % (
-                i,
-                file1,
-                ", ".join(x1.keys()),
-                file2,
-                ", ".join(x2.keys()),
-            )
-        )
-        results.append({})
-        for k in x1.keys():
-            v1 = x1[k].numpy().astype(np.double)
-            v2 = x2[k].numpy().astype(np.double)
-            mae = abs(v1 - v2).mean()
-            results[-1][k] = mae
-
-    total = sum(sum(x.values()) for x in results)
-    count = sum(len(x.values()) for x in results)
-    mean = total / count
-    return results, mean
-
-
 def diff_stats(file1, file2, per_tensor_and_channel=False):
     dataset1 = NumpyTFReader(file1)
     dataset2 = NumpyTFReader(file2)