Python 断言异常

Question

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

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

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

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

我正在尝试下面的代码。

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

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

谢谢。

Answer 1

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

assert count > 0

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

===

回答以下问题：

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

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

Python Assert exception