Fix tests

PostHog · Mar 21, 2024 · f31a9e7 · f31a9e7
1 parent fb007ca
commit f31a9e7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/posthog/hogql/test/test_bytecode.py b/posthog/hogql/test/test_bytecode.py
@@ -130,7 +130,7 @@ def test_bytecode_create(self):
     def test_bytecode_create_error(self):
         with self.assertRaises(NotImplementedException) as e:
             to_bytecode("(select 1)")
-        self.assertEqual(str(e.exception), "Visitor has no method visit_select_query")
+        self.assertEqual(str(e.exception), "BytecodeBuilder has no method visit_select_query")
 
         with self.assertRaises(NotImplementedException) as e:
             to_bytecode("1 in cohort 2")

diff --git a/posthog/hogql/test/test_visitor.py b/posthog/hogql/test/test_visitor.py
@@ -125,7 +125,7 @@ def visit_arithmetic_operation(self, node: ast.ArithmeticOperation):
 
         with self.assertRaises(HogQLException) as e:
             UnknownNotDefinedVisitor().visit(parse_expr("1 + 3 / 'asd2'"))
-        self.assertEqual(str(e.exception), "Visitor has no method visit_constant")
+        self.assertEqual(str(e.exception), "UnknownNotDefinedVisitor has no method visit_constant")
 
     def test_hogql_exception_start_end(self):
         class EternalVisitor(TraversingVisitor):