# Path: morph_net/framework/op_handlers.py
# RESIZE_OP_NAMES = [
#     'ResizeArea', 'ResizeBicubic', 'ResizeBilinear', 'ResizeNearestNeighbor'
# ]
# def _get_base_op_hander_dicts():
# def get_gamma_op_handler_dict():
# def get_group_lasso_op_handler_dict():


from morph_net.framework import op_handlers
import tensorflow.compat.v1 as tf
"""Tests for framework.op_handlers."""

from __future__ import absolute_import
from __future__ import division
from __future__ import print_function




class OpHandlersTest(tf.test.TestCase):

  def test_dict_logic(self):
    gamma_dict = op_handlers.get_gamma_op_handler_dict()
    self.assertIn('Conv2D', gamma_dict)
    self.assertIn('MatMul', gamma_dict)

# Path: morph_net/framework/op_handlers.py
# RESIZE_OP_NAMES = [
#     'ResizeArea', 'ResizeBicubic', 'ResizeBilinear', 'ResizeNearestNeighbor'
# ]
# def _get_base_op_hander_dicts():
# def get_gamma_op_handler_dict():
# def get_group_lasso_op_handler_dict():


# Path: morph_net/framework/op_handlers_test.py
from morph_net.framework import op_handlers
import tensorflow.compat.v1 as tf

"""Tests for framework.op_handlers."""

from __future__ import absolute_import
from __future__ import division
from __future__ import print_function




class OpHandlersTest(tf.test.TestCase):

  def test_dict_logic(self):
    gamma_dict = op_handlers.get_gamma_op_handler_dict()
    self.assertIn('Conv2D', gamma_dict)
    self.assertIn('MatMul', gamma_dict)

    group_lasso_dict = op_handlers.get_group_lasso_op_handler_dict()