125 lines
2.4 KiB
Python
125 lines
2.4 KiB
Python
from rolltable import tables
|
|
|
|
fixture_metadata = """
|
|
metadata:
|
|
headers:
|
|
- Header 1
|
|
- Header 2
|
|
- Header 3
|
|
die: 10
|
|
frequencies:
|
|
default:
|
|
Option 1: 0.3
|
|
Option 2: 0.5
|
|
Option 3: 0.2
|
|
nondefault:
|
|
Option 1: 0.0
|
|
Option 2: 0.1
|
|
Option 3: 0.9
|
|
"""
|
|
|
|
fixture_source = """
|
|
Option 1:
|
|
- choice 1: description 1
|
|
- choice 2: description 2
|
|
- choice 3: description 3
|
|
Option 2:
|
|
- choice 1: description 4
|
|
- choice 2: description 5
|
|
- choice 3: description 6
|
|
Option 3:
|
|
- choice 1: description 7
|
|
- choice 2: description 8
|
|
- choice 3: description 9
|
|
"""
|
|
|
|
fixture_one_choice = """
|
|
option 1:
|
|
- choice 1: description 1
|
|
"""
|
|
|
|
fixture_repeated_choices = """
|
|
option 1:
|
|
- choice 1: description 1
|
|
- choice 1: description 1
|
|
- choice 1: description 1
|
|
"""
|
|
|
|
fixture_no_descriptions = """
|
|
option 1:
|
|
- choice 1
|
|
"""
|
|
|
|
fixture_combined_A = """
|
|
A1:
|
|
- A choice 1
|
|
- A choice 2
|
|
- A choice 3
|
|
A2:
|
|
- A choice 4
|
|
- A choice 5
|
|
- A choice 6
|
|
A3:
|
|
- A choice 7
|
|
- A choice 8
|
|
- A choice 9
|
|
"""
|
|
|
|
fixture_combined_B = """
|
|
metadata:
|
|
headers:
|
|
- HeaderB
|
|
- HeaderB_Choice
|
|
B1:
|
|
- B choice 1
|
|
B2:
|
|
- B choice 2
|
|
B3:
|
|
- B choice 3
|
|
"""
|
|
|
|
|
|
def test_combined_tables():
|
|
tA = tables.RollTable(fixture_combined_A)
|
|
tB = tables.RollTable(fixture_combined_B)
|
|
|
|
combined = tables.CombinedTable(tables=[tA, tB], die=6)
|
|
assert 'A1' in str(combined)
|
|
assert 'B1' in str(combined)
|
|
|
|
|
|
def test_table_end_to_end():
|
|
assert str(tables.RollTable(fixture_source))
|
|
|
|
|
|
def test_table_end_to_end_with_metadata():
|
|
assert str(tables.RollTable(fixture_metadata + fixture_source))
|
|
|
|
|
|
def test_table_frequency():
|
|
t = tables.RollTable(fixture_metadata + fixture_source, frequency='nondefault')
|
|
assert t.frequencies['Option 1'] == 0.0
|
|
assert t.frequencies['Option 2'] == 0.1
|
|
assert t.frequencies['Option 3'] == 0.9
|
|
|
|
|
|
def test_one_option():
|
|
t = tables.RollTable(fixture_one_choice, die=1)
|
|
assert t.values == [('option 1', {'choice 1': 'description 1'})]
|
|
|
|
|
|
def test_collapsed():
|
|
t = tables.RollTable(fixture_repeated_choices, die=6, collapsed=True)
|
|
assert len(list(t.rows)) == 1
|
|
|
|
|
|
def test_not_collapsed():
|
|
t = tables.RollTable(fixture_repeated_choices, die=6, collapsed=False)
|
|
assert len(list(t.rows)) == 6
|
|
|
|
|
|
def test_no_descriptions():
|
|
t = tables.RollTable(fixture_no_descriptions, die=1)
|
|
assert 'd1' in str(t)
|
|
assert 'option 1' in str(t)
|