from tensorflow.compiler.mlir.tfr.python import test_utils
from tensorflow.python.framework import test_ops
from tensorflow.python.platform import test


class TestOpsDefsTest(test_utils.OpsDefsTest):

  def test_test_ops(self):
    attr = tf.function(test_ops.test_attr)(tf.float32)
    self.assertAllClose(attr.numpy(), 100.0)


if __name__ == '__main__':
  os.environ['TF_MLIR_TFR_LIB_DIR'] = (

        def testTestOps = operations.children(firstLevelTestOps[0], ExecuteTestBuildOperationType)
        assert testTestOps.size() == 2
        assert testTestOps*.details.testDescriptor.name as Set == ["ok", "fail"] as Set
        assert testTestOps*.details.testDescriptor.className as Set == ["org.gradle.Test", "org.gradle.Test"] as Set
        assert testTestOps*.details.testDescriptor.composite == [false, false]