Function test_csv_serialises_correctly

tests/test_Statistics.py:7–30

Tests that writing a CSV of a ``Statistics`` object and then reading that CSV again returns the same object.

(ok_small, tmp_path)

def test_csv_serialises_correctly(ok_small, tmp_path):
    """
    Tests that writing a CSV of a ``Statistics`` object and then reading that
    CSV again returns the same object.
    """
    sol = Solution(ok_small, [[1, 2], [3, 4]])
    cost_eval = CostEvaluator([20], 6, 6)

    collected_stats = Statistics()
    for idx in range(10):  # populate the statistics object
        collected_stats.collect(sol, sol, sol, cost_eval)

    csv_path = tmp_path / "test.csv"
    assert_(not csv_path.exists())

    # Write the collected statistics to the CSV file location, and check that
    # the file now does exist.
    collected_stats.to_csv(csv_path)
    assert_(csv_path.exists())

    # Now read them back in, and check that the newly read object is the same
    # as the previously written object.
    read_stats = Statistics.from_csv(csv_path)
    assert_equal(collected_stats, read_stats)


@pytest.mark.parametrize("num_iterations", [0, 5])