Skip to content

Commit 3c24ddb

Browse files
author
cescofran
committed
Use simulations fixtures in test_simulation
1 parent 0b9d04c commit 3c24ddb

File tree

1 file changed

+5
-8
lines changed

1 file changed

+5
-8
lines changed

tests/core/test_simulations.py

Lines changed: 5 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
11
# -*- coding: utf-8 -*-
22

3-
def test_calculate_full_tracer(tax_benefit_system, simulation_builder):
4-
simulation = simulation_builder.build_default_simulation(tax_benefit_system)
3+
def test_calculate_full_tracer(simulation):
54
simulation.trace = True
65
simulation.calculate('income_tax', '2017-01')
76

@@ -21,8 +20,7 @@ def test_calculate_full_tracer(tax_benefit_system, simulation_builder):
2120
assert income_tax_node.parameters[0].value == 0.15
2221

2322

24-
def test_get_entity_not_found(tax_benefit_system, simulation_builder):
25-
simulation = simulation_builder.build_default_simulation(tax_benefit_system)
23+
def test_get_entity_not_found(simulation):
2624
assert simulation.get_entity(plural = "no_such_entities") is None
2725

2826

@@ -55,9 +53,8 @@ def test_clone(tax_benefit_system, simulation_builder):
5553
assert salary_holder_clone.population == simulation_clone.persons
5654

5755

58-
def test_get_memory_usage(tax_benefit_system, simulation_builder, single):
59-
simulation = simulation_builder.build_from_entities(tax_benefit_system, single)
60-
simulation.calculate('disposable_income', '2017-01')
61-
memory_usage = simulation.get_memory_usage(variables = ['salary'])
56+
def test_get_memory_usage(simulation_single):
57+
simulation_single.calculate('disposable_income', '2017-01')
58+
memory_usage = simulation_single.get_memory_usage(variables = ['salary'])
6259
assert(memory_usage['total_nb_bytes'] > 0)
6360
assert(len(memory_usage['by_variable']) == 1)

0 commit comments

Comments
 (0)