Python 断言异常

Question

我在一个数据块笔记本的 10 个数据块单元格中有 phyton 代码。

第一个单元格包含以下代码。

df = spark.sql(f"""select * from test.table """)
count= df.count()

这里如果 count 为 0 ，我希望笔记本失败，如果 count 大于 1 ，我希望 databricks 笔记本中的其余单元格执行。

我正在尝试下面的代码。

 df = spark.sql(f"""select * from test.table """)
    count= df.count()
    assert count <= 0

上面的代码没有引发断言错误，而是说查询不返回任何结果并执行数据块中的下一个单元格。有人可以让我知道引发断言异常并终止笔记本而不执行更多单元格。

谢谢。

Answer 1

你的断言倒退了。你断言你需要什么 是真的。

assert count > 0

（第一行的val 是什么？这是剪切粘贴错误吗？）

===

回答以下问题：

if count <= 0:
    raise AssertionError("must return at least one record")

您可以查看list of built-in Python 异常，看看其中一个是否比一般的“断言错误”更适合您的情况。