-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathtest_distribute_proceeds_for_krakatoa_ventures.py
121 lines (101 loc) · 4.67 KB
/
test_distribute_proceeds_for_krakatoa_ventures.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
import unittest
from distribute_proceeds_for_krakatoa_ventures import (
distribute_proceeds_for_krakatoa_ventures,
)
class TestDistributeProceedsForKrakatoaVentures(unittest.TestCase):
def setUp(self):
self.cap_table = {
"A": {
"total_membership_units_held": 10,
"unit_holders": [
{"name": "David", "units_held": 10, "amount_invested": 0}
],
},
"B": {
"total_membership_units_held": 20,
"unit_holders": [
{"name": "Alex", "units_held": 10, "amount_invested": 25000},
{"name": "Becky", "units_held": 10, "amount_invested": 25000},
],
},
"C": {
"total_membership_units_held": 5,
"unit_holders": [
{"name": "Becky", "units_held": 5, "amount_invested": 0}
],
},
}
def test_distribute_0_dollars(self):
cash_proceeds = 0
(
class_distribution,
member_distribution,
) = distribute_proceeds_for_krakatoa_ventures(cash_proceeds, self.cap_table)
class_distribution_total = sum(class_distribution.values())
member_distribution_total = sum(member_distribution.values())
self.assertDictEqual(dict(class_distribution), {"A": 0, "B": 0, "C": 0})
self.assertAlmostEqual(class_distribution_total, cash_proceeds, delta=1)
self.assertDictEqual(
dict(member_distribution), {"Alex": 0, "Becky": 0, "David": 0}
)
self.assertAlmostEqual(member_distribution_total, cash_proceeds, delta=1)
def test_distribute_100_dollars(self):
cash_proceeds = 10000
(
class_distribution,
member_distribution,
) = distribute_proceeds_for_krakatoa_ventures(cash_proceeds, self.cap_table)
class_distribution_total = sum(class_distribution.values())
member_distribution_total = sum(member_distribution.values())
self.assertDictEqual(dict(class_distribution), {"A": 0, "B": 10000, "C": 0})
self.assertAlmostEqual(class_distribution_total, cash_proceeds, delta=1)
self.assertDictEqual(
dict(member_distribution), {"Alex": 5000, "Becky": 5000, "David": 0}
)
self.assertAlmostEqual(member_distribution_total, cash_proceeds, delta=1)
def test_distribute_250_dollars_and_50_cents(self):
cash_proceeds = 25050
(
class_distribution,
member_distribution,
) = distribute_proceeds_for_krakatoa_ventures(cash_proceeds, self.cap_table)
class_distribution_total = sum(class_distribution.values())
member_distribution_total = sum(member_distribution.values())
self.assertDictEqual(dict(class_distribution), {"A": 0, "B": 25050, "C": 0})
self.assertAlmostEqual(class_distribution_total, cash_proceeds, delta=1)
self.assertDictEqual(
dict(member_distribution), {"Alex": 12525, "Becky": 12525, "David": 0}
)
self.assertAlmostEqual(member_distribution_total, cash_proceeds, delta=1)
def test_distribute_500_dollars(self):
cash_proceeds = 50000
(
class_distribution,
member_distribution,
) = distribute_proceeds_for_krakatoa_ventures(cash_proceeds, self.cap_table)
class_distribution_total = sum(class_distribution.values())
member_distribution_total = sum(member_distribution.values())
self.assertDictEqual(dict(class_distribution), {"A": 0, "B": 50000, "C": 0})
self.assertAlmostEqual(class_distribution_total, cash_proceeds, delta=1)
self.assertDictEqual(
dict(member_distribution), {"Alex": 25000, "Becky": 25000, "David": 0}
)
self.assertAlmostEqual(member_distribution_total, cash_proceeds, delta=1)
def test_distribute_1000_dollars(self):
cash_proceeds = 100000
(
class_distribution,
member_distribution,
) = distribute_proceeds_for_krakatoa_ventures(cash_proceeds, self.cap_table)
class_distribution_total = sum(class_distribution.values())
member_distribution_total = sum(member_distribution.values())
self.assertDictEqual(
dict(class_distribution), {"A": 14286, "B": 78572, "C": 7143}
)
self.assertAlmostEqual(class_distribution_total, cash_proceeds, delta=1)
self.assertDictEqual(
dict(member_distribution), {"Alex": 39286, "Becky": 46429, "David": 14286}
)
self.assertAlmostEqual(member_distribution_total, cash_proceeds, delta=1)
if __name__ == "__main__":
unittest.main()