using ::testing::NotNull;
using ::testing::SizeIs;

// Fixture for testing TfSavedModel functionalities. Initializes a
// `MLIRContext` by loading the `tf_saved_model` dialect.
class TfSavedModelTest : public ::testing::Test {
 protected:
  TfSavedModelTest() : ctx_() {
    ctx_.loadDialect<TensorFlowSavedModelDialect, func::FuncDialect>();
  }

  MLIRContext ctx_;
};

// Parses `module_op_str` and returns the resulting `ModuleOp`.