75 lines · cpp
1//===- CharacterTest.cpp -- CharacterExprHelper unit tests ----------------===//2//3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.4// See https://llvm.org/LICENSE.txt for license information.5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception6//7//===----------------------------------------------------------------------===//8 9#include "flang/Optimizer/Builder/Character.h"10#include "gtest/gtest.h"11#include "flang/Optimizer/Builder/BoxValue.h"12#include "flang/Optimizer/Builder/FIRBuilder.h"13#include "flang/Optimizer/Dialect/Support/KindMapping.h"14#include "flang/Optimizer/Support/InitFIR.h"15 16struct CharacterTest : public testing::Test {17public:18 void SetUp() override {19 fir::support::loadDialects(context);20 21 kindMap = std::make_unique<fir::KindMapping>(&context,22 "i10:80,l3:24,a1:8,r54:Double,c20:X86_FP80,r11:PPC_FP128,"23 "r12:FP128,r13:X86_FP80,r14:Double,r15:Float,r16:Half,r23:BFloat");24 mlir::OpBuilder builder(&context);25 auto loc = builder.getUnknownLoc();26 27 // Set up a Module with a dummy function operation inside.28 // Set the insertion point in the function entry block.29 moduleOp = mlir::ModuleOp::create(builder, loc);30 builder.setInsertionPointToStart(moduleOp->getBody());31 mlir::func::FuncOp func = mlir::func::FuncOp::create(32 builder, loc, "func1", builder.getFunctionType({}, {}));33 auto *entryBlock = func.addEntryBlock();34 builder.setInsertionPointToStart(entryBlock);35 36 firBuilder = std::make_unique<fir::FirOpBuilder>(builder, *kindMap);37 }38 39 fir::FirOpBuilder &getBuilder() { return *firBuilder; }40 41 mlir::MLIRContext context;42 mlir::OwningOpRef<mlir::ModuleOp> moduleOp;43 std::unique_ptr<fir::KindMapping> kindMap;44 std::unique_ptr<fir::FirOpBuilder> firBuilder;45};46 47TEST_F(CharacterTest, smallUtilityFunctions) {48 auto builder = getBuilder();49 auto loc = builder.getUnknownLoc();50 llvm::StringRef strValue("onestringliteral");51 auto strLit = fir::factory::createStringLiteral(builder, loc, strValue);52 EXPECT_TRUE(53 fir::factory::CharacterExprHelper::hasConstantLengthInType(strLit));54 auto ty = strLit.getCharBox()->getAddr().getType();55 EXPECT_TRUE(fir::factory::CharacterExprHelper::isCharacterScalar(ty));56 EXPECT_EQ(fir::factory::CharacterExprHelper::getCharacterOrSequenceKind(ty),57 fir::factory::CharacterExprHelper::getCharacterKind(ty));58}59 60TEST_F(CharacterTest, createSubstring) {61 auto builder = getBuilder();62 auto loc = builder.getUnknownLoc();63 auto charHelper = fir::factory::CharacterExprHelper(builder, loc);64 llvm::StringRef data("a dummy string to test substring");65 auto str = fir::factory::createStringLiteral(builder, loc, data);66 auto lb = builder.createIntegerConstant(loc, builder.getI64Type(), 18);67 auto ub = builder.createIntegerConstant(loc, builder.getI64Type(), 22);68 auto substr = charHelper.createSubstring(*str.getCharBox(), {lb, ub});69 EXPECT_FALSE(70 fir::factory::CharacterExprHelper::hasConstantLengthInType(substr));71 EXPECT_FALSE(charHelper.getCharacterType(substr).hasConstantLen());72 EXPECT_FALSE(fir::factory::CharacterExprHelper::isArray(73 charHelper.getCharacterType(substr)));74}75