下面是我的主要代码，我想单元测试
获取数据.py

from pyspark.sql import SparkSession
from pyspark_llap.sql.session import HiveWarehouseSession

def get_hive_data(query):
        hive_data = hive.executeQuery(query)
        return hive_data

if __name__ == "__main__":
    spark = SparkSession\
        .builder\
        .appName("HiveApp")\
        .getOrCreate()

    hive=HiveWarehouseSession.session(spark).build()
    data = get_hive_data()

下面是我的unittest代码，我只在这里编写了导入，因为我从get\u数据导入时出错/*/
测试.py

import unittest
import pyspark

import pyspark.sql.functions as f
from pyspark.sql import functions as F
from pyspark.sql import SparkSession
from get_data import *

错误：没有名为pyspark\u llap的模块
但是如果我运行just get\ u data.py，它将成功运行，我正在边缘节点上运行它！