Skip to content

Commit

Permalink
a few more tests - various charts
Browse files Browse the repository at this point in the history
  • Loading branch information
ADBond committed Sep 12, 2024
1 parent a73ae05 commit 5fb7dee
Showing 1 changed file with 28 additions and 0 deletions.
28 changes: 28 additions & 0 deletions tests/test_basic_functionality.py
Original file line number Diff line number Diff line change
Expand Up @@ -87,3 +87,31 @@ def test_completeness(api_info, fake_1000_factory):
df = fake_1000_factory(api_info["version"])

completeness_chart(df, db_api=db_api)


def test_match_weights_chart(api_info, fake_1000_factory, fake_1000_settings):
db_api = api_info["db_api"]
df = fake_1000_factory(api_info["version"])
linker = Linker(df, fake_1000_settings, db_api)
linker.visualisations.match_weights_chart()


def test_parameter_estimates_chart(api_info, fake_1000_factory, fake_1000_settings):
db_api = api_info["db_api"]
df = fake_1000_factory(api_info["version"])
linker = Linker(df, fake_1000_settings, db_api)
linker.training.estimate_parameters_using_expectation_maximisation(
block_on("dob"),
)
linker.training.estimate_parameters_using_expectation_maximisation(
block_on("first_name", "surname"),
)
linker.visualisations.parameter_estimate_comparisons_chart()


def test_m_u_chart(api_info, fake_1000_factory, fake_1000_settings):
db_api = api_info["db_api"]
df = fake_1000_factory(api_info["version"])
linker = Linker(df, fake_1000_settings, db_api)

linker.visualisations.m_u_parameters_chart()

0 comments on commit 5fb7dee

Please sign in to comment.