import sys, os

sys.path.append(os.path.dirname(os.getcwd()))

from tensorflow import keras
import numpy as np
from pyradox import convnets


def test():
    inputs = keras.Input(shape=(28, 28, 1))
    x = keras.layers.ZeroPadding2D(2)(
        inputs
    )  # padding to increase dimenstions to 32x32
    x = keras.layers.Conv2D(3, 1, padding="same")(
        x
    )  # increasing the number of channels to 3
    x = convnets.ResNeXt([(32, 3), (64, 5), (32, 5), (64, 3)])(x)
    x = keras.layers.GlobalAvgPool2D()(x)
    outputs = keras.layers.Dense(10, activation="softmax")(x)

    model = keras.models.Model(inputs=inputs, outputs=outputs)

tensorflow.keras.layers.GlobalAvgPool2D tensorflow.keras.Input tensorflow.keras.models.Model tensorflow.keras.layers.Dense tensorflow.keras.layers.Conv2D tensorflow.keras.layers.ZeroPadding2D
[(11, 'tensorflow.keras.Input', 'keras.Input', ([], {'shape': '(28, 28, 1)'}), False, 'from tensorflow import keras\n'), (22, 'tensorflow.keras.models.Model', 'keras.models.Model', ([], {'inputs': 'inputs', 'outputs': 'outputs'}), False, 'from tensorflow import keras\n'), (3, 'os.getcwd', 'os.getcwd', ([], {}), False, 'import sys, os\n'), (12, 'tensorflow.keras.layers.ZeroPadding2D', 'keras.layers.ZeroPadding2D', (['(2)'], {}), False, 'from tensorflow import keras\n'), (15, 'tensorflow.keras.layers.Conv2D', 'keras.layers.Conv2D', (['(3)', '(1)'], {'padding': '"""same"""'}), False, 'from tensorflow import keras\n'), (18, 'pyradox.convnets.ResNeXt', 'convnets.ResNeXt', (['[(32, 3), (64, 5), (32, 5), (64, 3)]'], {}), False, 'from pyradox import convnets\n'), (19, 'tensorflow.keras.layers.GlobalAvgPool2D', 'keras.layers.GlobalAvgPool2D', ([], {}), False, 'from tensorflow import keras\n'), (20, 'tensorflow.keras.layers.Dense', 'keras.layers.Dense', (['(10)'], {'activation': '"""softmax"""'}), False, 'from tensorflow import keras\n')]