b.scala

import org.apache.spark.sql.types.{IntegerType, StringType, StructField, StructType}
import org.scalatest.funsuite.AnyFunSuite
import org.scalatest.matchers.should.Matchers

class MyDataFrameTest extends AnyFunSuite with Matchers {
  
  test("test schema of myDataFrame") {
    val myDataFrame = Seq(
      ("John", 25),
      ("Jane", 30),
      ("Bob", 40)
    ).toDF("name", "age")
    
    val expectedSchema = StructType(Seq(
      StructField("name", StringType, nullable = true),
      StructField("age", IntegerType, nullable = true)
    ))
    
    myDataFrame.schema shouldEqual expectedSchema
  }
  
}