Skip to content

Commit

Permalink
fix: lint issue
Browse files Browse the repository at this point in the history
  • Loading branch information
arpitsrivastava-cb committed Dec 19, 2024
1 parent 78b5dff commit e0089fd
Show file tree
Hide file tree
Showing 5 changed files with 14 additions and 6 deletions.
2 changes: 1 addition & 1 deletion cdp/address.py
Original file line number Diff line number Diff line change
@@ -1,14 +1,14 @@
from collections.abc import Iterator
from decimal import Decimal

from cdp.address_reputation import AddressReputation
from cdp.asset import Asset
from cdp.balance import Balance
from cdp.balance_map import BalanceMap
from cdp.cdp import Cdp
from cdp.faucet_transaction import FaucetTransaction
from cdp.historical_balance import HistoricalBalance
from cdp.transaction import Transaction
from cdp.address_reputation import AddressReputation


class Address:
Expand Down
6 changes: 5 additions & 1 deletion cdp/address_reputation.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,3 @@
from cdp.cdp import Cdp
from cdp.client import AddressReputationMetadata
from cdp.client.models.address_reputation import AddressReputation as AddressReputationModel

Expand All @@ -16,19 +15,24 @@ def __init__(self, model: AddressReputationModel) -> None:

@property
def metadata(self) -> AddressReputationMetadata:
"""Return the metadata of the address."""
return self._metadata

@property
def score(self) -> int:
"""Return the score of the address."""
return self._score

@property
def risky(self) -> bool:
"""Return whether the address is risky."""
return self.score < 0

def __str__(self) -> str:
"""Return a string representation of the AddressReputation."""
metadata = ", ".join(f"{key}={getattr(self.metadata, key)}" for key in vars(self.metadata))
return f"Address Reputation: (score={self.score}, metadata=({metadata}))"

def __repr__(self) -> str:
"""Return a string representation of the AddressReputation."""
return str(self)
2 changes: 1 addition & 1 deletion tests/factories/address_reputation_factory.py
Original file line number Diff line number Diff line change
@@ -1,8 +1,8 @@
import pytest

from cdp.address_reputation import AddressReputation
from cdp.client.models.address_reputation import AddressReputation as AddressReputationModel
from cdp.client.models.address_reputation_metadata import AddressReputationMetadata
from cdp.address_reputation import AddressReputation


@pytest.fixture
Expand Down
2 changes: 1 addition & 1 deletion tests/test_address.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,13 +4,13 @@
import pytest

from cdp.address import Address
from cdp.address_reputation import AddressReputation
from cdp.balance_map import BalanceMap
from cdp.client.exceptions import ApiException
from cdp.errors import ApiError
from cdp.faucet_transaction import FaucetTransaction
from cdp.historical_balance import HistoricalBalance
from cdp.transaction import Transaction
from cdp.address_reputation import AddressReputation


def test_address_initialization(address_factory):
Expand Down
8 changes: 6 additions & 2 deletions tests/test_address_reputation.py
Original file line number Diff line number Diff line change
@@ -1,27 +1,29 @@
import pytest
from unittest.mock import Mock, patch
from cdp.address_reputation import AddressReputation


def test_address_reputation_initialization(address_reputation_factory):
"""Test address reputation initialization."""
address_reputation = address_reputation_factory()
assert isinstance(address_reputation, AddressReputation)


def test_address_reputation_score(address_reputation_factory):
"""Test address reputation score."""
address_reputation = address_reputation_factory()

assert address_reputation.score == 10


def test_address_reputation_metadata(address_reputation_factory):
"""Test address reputation metadata."""
address_reputation = address_reputation_factory()

assert address_reputation.metadata.total_transactions == 2
assert address_reputation.metadata.unique_days_active == 3


def test_address_reputation_risky(address_reputation_factory):
"""Test address reputation risky."""
address_reputation = address_reputation_factory(score=-5)
assert address_reputation.risky is True

Expand All @@ -33,12 +35,14 @@ def test_address_reputation_risky(address_reputation_factory):


def test_address_reputation_str(address_reputation_factory):
"""Test address reputation str."""
address_reputation = address_reputation_factory(score=10)
expected_str = "Address Reputation: (score=10, metadata=(total_transactions=2, unique_days_active=3, longest_active_streak=4, current_active_streak=5, activity_period_days=6, token_swaps_performed=7, bridge_transactions_performed=8, lend_borrow_stake_transactions=9, ens_contract_interactions=10, smart_contract_deployments=10))"
assert str(address_reputation) == expected_str


def test_address_reputation_repr(address_reputation_factory):
"""Test address reputation repr."""
address_reputation = address_reputation_factory(score=10)
expected_repr = "Address Reputation: (score=10, metadata=(total_transactions=2, unique_days_active=3, longest_active_streak=4, current_active_streak=5, activity_period_days=6, token_swaps_performed=7, bridge_transactions_performed=8, lend_borrow_stake_transactions=9, ens_contract_interactions=10, smart_contract_deployments=10))"
assert repr(address_reputation) == expected_repr

0 comments on commit e0089fd

Please sign in to comment.