fixed validation and test

industrydive · Nov 29, 2023 · b6d8a01 · b6d8a01
1 parent abcdb92
commit b6d8a01
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/bigquery_exporter/base.py b/bigquery_exporter/base.py
@@ -9,7 +9,7 @@ def custom_field(method):
  Decorator to mark a method as a custom field for a BigQueryExporter subclass.
  """
  # Ensure that the method has exactly two arguments: self and the Django model instance
- assert method.__code__.co_argcount != 2, \
+ assert method.__code__.co_argcount == 2, \
  'Custom field methods must have exactly two arguments: self and the Django model instance'
  method.is_custom_field = True
  return method

diff --git a/tests/test_base.py b/tests/test_base.py
@@ -80,7 +80,7 @@ def test_export_logs_error_on_exception(self, test_exporter, caplog):
 
  def test_custom_field_decorator_sets_custom_attribute_on_callable(self):
  @custom_field
- def test_field():
+ def test_field(self, obj):
  pass
 
  assert test_field.is_custom_field