Mimic code structure in random rewriting tests

bfeabc82 · Ricardo Vieira · Ricardo Vieira · e67dc97c · bfeabc82 · bfeabc82
--- a/tests/tensor/random/rewriting/__init__.py
+++ b/tests/tensor/random/rewriting/__init__.py
--- a/tests/tensor/random/test_rewriting.py
+++ b/tests/tensor/random/test_rewriting.py