Ejemplo n.º 1
0
class TestCollapse(object):
    """This has its own class becuase the setup is different."""

    @classmethod
    def setUp(self):
        self.dg = DebtGraph({"alice": {"charlie": 5}, "bob": {"alice": 10}, "charlie": {"bob": 10}})

    def test_collapse(self):
        """Collapsing the graph should work as intended."""
        self.dg.collapse()
        assert self.dg.graph == {"charlie": {"alice": 5.0}, "bob": {}, "alice": {}}
Ejemplo n.º 2
0
class TestEdit(object):
    """Testing functions that manipulate the list of people in the graph."""

    @classmethod
    def setUp(self):
        """Set up a test DebtGraph."""
        self.dg = DebtGraph({"alice": {"bob": 20, "charlie": 5}, "bob": {"alice": 10}, "charlie": {}})

    def test_add_person(self):
        """Should add a person to the graph correctly."""
        self.dg.add_person("danielle")
        assert self.dg.graph == {
            "alice": {"bob": 20, "charlie": 5},
            "bob": {"alice": 10},
            "charlie": {},
            "danielle": {},
        }

    def test_remove_person_normal(self):
        """Should remove a person and all their debts correctly."""
        self.dg.remove_person("alice")
        assert self.dg.graph == {"bob": {}, "charlie": {}}

    @raises(KeyError)
    def test_remove_person_error(self):
        """Remove person should raise an error if the person does not exist."""
        self.dg.remove_person("zack")

    def test_rename_person_normal(self):
        """Should rename a person correctly when they exist."""
        self.dg.rename_person("alice", "alex")
        assert self.dg.graph == {"alex": {"bob": 20, "charlie": 5}, "bob": {"alex": 10}, "charlie": {}}

    @raises(KeyError)
    def test_rename_person_error(self):
        """Rename person should throw an error if the person does not exist."""
        self.dg.rename_person("zack", "anthony")
Ejemplo n.º 3
0
 def test_addition_with_ints(self):
     """Addition should work when the keys are ints."""
     self.dg = DebtGraph({1: {2: 20, 3: 5}, 2: {1: 10}, 3: {}})
     self.dg.add(1, 2, 5)
     assert self.dg.graph == {1: {2: 25, 3: 5}, 2: {1: 10}, 3: {}}
Ejemplo n.º 4
0
 def setUp(self):
     """Set up a test DebtGraph."""
     self.dg = DebtGraph({"alice": {"bob": 20, "charlie": 5}, "bob": {"alice": 10}, "charlie": {}})
Ejemplo n.º 5
0
class TestManipulation(object):
    """Graph manipulation methods should work properly."""

    @classmethod
    def setUp(self):
        """Set up a test DebtGraph."""
        self.dg = DebtGraph({"alice": {"bob": 20, "charlie": 5}, "bob": {"alice": 10}, "charlie": {}})

    # removal

    def test_remove_normal(self):
        """Should remove debt correctly."""
        self.dg.remove("alice", "bob", 5)
        assert self.dg.graph == {"alice": {"bob": 15, "charlie": 5}, "bob": {"alice": 10}, "charlie": {}}

    def test_remove_deletion(self):
        """Should delete the node if more debt is removed than exists."""
        self.dg.remove("alice", "bob", 25)
        assert self.dg.graph == {"alice": {"charlie": 5}, "bob": {"alice": 10}, "charlie": {}}

    @raises(KeyError)
    def test_remove_error(self):
        """Should throw an error when trying to remove a nonexistant debt."""
        self.dg.remove("charlie", "bob", 4)

    # forgiveness

    def test_forgive_normal(self):
        """Forgive method should delete debts."""
        self.dg.forgive("bob", "alice")
        assert self.dg.graph == {"alice": {"bob": 20, "charlie": 5}, "bob": {}, "charlie": {}}

    @raises(KeyError)
    def test_forgive_error(self):
        """Forgive method should throw an error when used on a nonexistant debt."""
        self.dg.forgive("bob", "charlie")

    # addition

    def test_addition_normal(self):
        """Adding to an already existing debt should increase it."""
        self.dg.add("alice", "bob", 5)
        assert self.dg.graph == {"alice": {"bob": 25, "charlie": 5}, "bob": {"alice": 10}, "charlie": {}}

    def test_addition_split(self):
        """Adding a debt with multiple debtors should split the debt."""
        self.dg.add("charlie", ["alice", "bob"], 10)
        assert self.dg.graph == {
            "alice": {"bob": 20, "charlie": 5},
            "bob": {"alice": 10},
            "charlie": {"alice": 5, "bob": 5},
        }

    def test_addition_split_self(self):
        """Adding a debt with multiple debtors including the creditor should
        spilt the debt between everyone except the creditor."""
        self.dg.add("charlie", ["charlie", "alice", "bob"], 15)
        assert self.dg.graph == {
            "alice": {"bob": 20, "charlie": 5},
            "bob": {"alice": 10},
            "charlie": {"alice": 5, "bob": 5},
        }

    def test_addition_with_ints(self):
        """Addition should work when the keys are ints."""
        self.dg = DebtGraph({1: {2: 20, 3: 5}, 2: {1: 10}, 3: {}})
        self.dg.add(1, 2, 5)
        assert self.dg.graph == {1: {2: 25, 3: 5}, 2: {1: 10}, 3: {}}

    @raises(TypeError)
    def test_addition_error1(self):
        """Addition should throw an error if not called using a list, int, or
        string."""
        self.dg.add("alice", {}, 3)

    def test_addition_error2(self):
        """Addition should throw an error if any of the people
        specified don't exist."""
        assert_raises(KeyError, lambda: self.dg.add("alice", "not here", 3))
        assert_raises(KeyError, lambda: self.dg.add("not here", "alice", 3))
        assert_raises(KeyError, lambda: self.dg.add("alice", ["bob", "NH"], 3))

    # cancellation

    def test_cancel(self):
        """Canceling useless debts should produce the correct graph."""
        self.dg.cancel()
        print self.dg.graph
        assert self.dg.graph == {"alice": {"bob": 10, "charlie": 5}, "bob": {}, "charlie": {}}
Ejemplo n.º 6
0
 def setUp(self):
     self.dg = DebtGraph({"alice": {"charlie": 5}, "bob": {"alice": 10}, "charlie": {"bob": 10}})