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-exception 6 // 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 16 struct CharacterTest : public testing::Test { 17 public: 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 = builder.create<mlir::ModuleOp>(loc); 30 builder.setInsertionPointToStart(moduleOp->getBody()); 31 mlir::func::FuncOp func = builder.create<mlir::func::FuncOp>( 32 loc, "func1", builder.getFunctionType(std::nullopt, std::nullopt)); 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 47 TEST_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 60 TEST_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