Added more tests

Author: fieryhenry

requirements_dev.txt

@@ -0,0 +1,5 @@
+flake8==3.9.2
+tox==3.24.3
+pytest==6.2.5
+pytest-cov==2.12.1
+mypy==0.910

tests/__init__.py

@@ -1 +1 @@
-from . import test_parse
+from . import test_item, test_parse, test_edits

tests/test_edits/__init__.py

@@ -0,0 +1 @@
+from . import test_basic, test_cats

tests/test_edits/test_basic/__init__.py

@@ -0,0 +1 @@
+from . import test_talent_orbs, test_basic

tests/test_edits/test_basic/test_basic.py

@@ -0,0 +1,40 @@
+"""Test basic item editing"""
+
+from pytest import MonkeyPatch
+from BCSFE_Python.edits.basic import basic_items
+
+
+def test_cf_normal(monkeypatch: MonkeyPatch):
+    """Test that the value is set correctly"""
+
+    inputs = ["y", "10"]
+    monkeypatch.setattr("builtins.input", lambda: inputs.pop(0))
+
+    save_stats = {"cat_food": {"Value": 50}}
+
+    save_stats = basic_items.edit_cat_food(save_stats)
+    assert save_stats["cat_food"]["Value"] == 10
+
+
+def test_catfood_leave(monkeypatch: MonkeyPatch):
+    """Test that the value is left unchanged"""
+
+    inputs = ["n", "10"]
+    monkeypatch.setattr("builtins.input", lambda: inputs.pop(0))
+
+    save_stats = {"cat_food": {"Value": 50}}
+
+    save_stats = basic_items.edit_cat_food(save_stats)
+    assert save_stats["cat_food"]["Value"] == 50
+
+
+def test_iq_normal(monkeypatch: MonkeyPatch):
+    """Test that the value is set correctly"""
+
+    inputs = ["abcdefghi"]
+    monkeypatch.setattr("builtins.input", lambda: inputs.pop(0))
+
+    save_stats = {"inquiry_code": "123456789"}
+
+    save_stats = basic_items.edit_inquiry_code(save_stats)
+    assert save_stats["inquiry_code"] == "abcdefghi"

tests/test_edits/test_basic/test_talent_orbs.py

@@ -0,0 +1,63 @@
+"""Test talent orbs"""
+
+from typing import Any
+from pytest import MonkeyPatch
+
+from BCSFE_Python.edits.basic import talent_orbs
+
+
+def test_create_orb_list():
+    """Test that the list of all possible talent orbs is created correctly"""
+
+    types = talent_orbs.get_talent_orbs_types()
+    assert len(types) == 155
+
+    assert "Red D Attack" == types[0]
+    assert "Red C Attack" == types[1]
+
+    assert "Red D Defense" == types[5]
+
+    assert "Floating D Attack" == types[10]
+
+    assert "Red D Strong" == types[65]
+    assert "Alien S Resistant" == types[139]
+
+
+def test_edit_all_orbs(monkeypatch: MonkeyPatch):
+    """Test that all talent orbs are edited correctly"""
+
+    inputs = ["10"]
+    monkeypatch.setattr("builtins.input", lambda: inputs.pop(0))
+    save_stats: dict[str, Any] = {
+        "talent_orbs": {
+            51: 5,
+            0: 10,
+            98: 0,
+            19: 1,
+        }
+    }
+    types = talent_orbs.get_talent_orbs_types()
+    save_stats = talent_orbs.edit_all_orbs(save_stats, types)
+
+    assert len(save_stats["talent_orbs"]) == len(types)
+    for i in range(len(types)):
+        assert save_stats["talent_orbs"][i] == 10
+
+
+def test_edit_all_orbs_invalid(monkeypatch: MonkeyPatch):
+    """Test that all talent orbs are edited correctly"""
+
+    inputs = ["aaa", "15"]
+    monkeypatch.setattr("builtins.input", lambda: inputs.pop(0))
+    save_stats_b: dict[str, Any] = {
+        "talent_orbs": {
+            51: 5,
+            0: 10,
+            98: 0,
+            19: 1,
+        }
+    }
+    types = talent_orbs.get_talent_orbs_types()
+    save_stats = talent_orbs.edit_all_orbs(save_stats_b, types)
+
+    assert save_stats_b == save_stats

tests/test_edits/test_cats/__init__.py

@@ -0,0 +1 @@
+from . import test_cat_id_selector

tests/test_edits/test_cats/test_cat_id_selector.py

@@ -0,0 +1,177 @@
+from pytest import MonkeyPatch
+from BCSFE_Python.edits.cats import cat_id_selector
+
+
+def test_get_all_cats():
+    """Test that the ids are correct"""
+    save_stats = {"cats": [1, 1, 1, 0, 1]}
+    ids = cat_id_selector.get_all_cats(save_stats)
+    assert ids == [0, 1, 2, 3, 4]
+
+
+def test_select_range(monkeypatch: MonkeyPatch):
+    """Test that the ids are correct"""
+
+    inputs = ["1-5"]
+    monkeypatch.setattr("builtins.input", lambda: inputs.pop(0))
+    save_stats = {"cats": [1, 1, 1, 0, 1]}
+
+    ids = cat_id_selector.select_cats_range(save_stats)
+    actual_ids = [1, 2, 3, 4, 5]
+    assert ids == actual_ids
+
+def test_select_current():
+    """Test that the ids are correct"""
+
+    save_stats = {"cats": [1, 1, 1, 0, 1]}
+
+    ids = cat_id_selector.select_current_cats(save_stats)
+    actual_ids = [0, 1, 2, 4]
+    assert ids == actual_ids
+
+def test_select_rarity(monkeypatch: MonkeyPatch):
+    """Test that the ids are correct"""
+
+    inputs = ["1"]
+    monkeypatch.setattr("builtins.input", lambda: inputs.pop(0))
+
+    ids = cat_id_selector.select_cats_rarity(False)
+    actual_ids = [0, 1, 2, 3, 4, 5, 6, 7, 8, 643]
+    assert ids == actual_ids
+
+def test_select_range_reverse(monkeypatch: MonkeyPatch):
+    """Test that the ids are correct"""
+
+    inputs = ["5-1"]
+    monkeypatch.setattr("builtins.input", lambda: inputs.pop(0))
+    save_stats = {"cats": [1, 1, 1, 0, 1]}
+
+    ids = cat_id_selector.select_cats_range(save_stats)
+    actual_ids = [1, 2, 3, 4, 5]
+    assert ids == actual_ids
+
+def test_select_range_mult(monkeypatch: MonkeyPatch):
+    """Test that the ids are correct"""
+
+    inputs = ["1-5 7-10"]
+    monkeypatch.setattr("builtins.input", lambda: inputs.pop(0))
+    save_stats = {"cats": [1, 1, 1, 0, 1]}
+
+    ids = cat_id_selector.select_cats_range(save_stats)
+    actual_ids = [1, 2, 3, 4, 5, 7, 8, 9, 10]
+    assert ids == actual_ids
+
+def test_select_range_spaces(monkeypatch: MonkeyPatch):
+    """Test that the ids are correct"""
+
+    inputs = ["1 4 7 2"]
+    monkeypatch.setattr("builtins.input", lambda: inputs.pop(0))
+    save_stats = {"cats": [1, 1, 1, 0, 1]}
+
+    ids = cat_id_selector.select_cats_range(save_stats)
+    actual_ids = [1, 4, 7, 2]
+    assert ids == actual_ids
+
+def test_select_range_comb(monkeypatch: MonkeyPatch):
+    """Test that the ids are correct"""
+
+    inputs = ["1 4 7 2 8-10"]
+    monkeypatch.setattr("builtins.input", lambda: inputs.pop(0))
+    save_stats = {"cats": [1, 1, 1, 0, 1]}
+
+    ids = cat_id_selector.select_cats_range(save_stats)
+    actual_ids = [1, 4, 7, 2, 8, 9, 10]
+    assert ids == actual_ids
+
+def test_select_range_all(monkeypatch: MonkeyPatch):
+    """Test that the ids are correct"""
+
+    inputs = ["all"]
+    monkeypatch.setattr("builtins.input", lambda: inputs.pop(0))
+    save_stats = {"cats": [1, 1, 1, 0, 1]}
+
+    ids = cat_id_selector.select_cats_range(save_stats)
+    actual_ids = [0, 1, 2, 3, 4]
+    assert ids == actual_ids
+
+
+def test_gatya_banner(monkeypatch: MonkeyPatch):
+    """Test that the ids are correct"""
+
+    inputs = ["602"]
+    monkeypatch.setattr("builtins.input", lambda: inputs.pop(0))
+
+    ids = cat_id_selector.select_cats_gatya_banner(False)
+    ids.sort()
+    actual_ids = [
+        448,
+        449,
+        450,
+        451,
+        455,
+        461,
+        463,
+        478,
+        481,
+        493,
+        544,
+        612,
+        138,
+        259,
+        330,
+        195,
+        496,
+        358,
+        376,
+        502,
+        526,
+        533,
+        564,
+        229,
+        30,
+        31,
+        32,
+        33,
+        35,
+        36,
+        39,
+        40,
+        61,
+        150,
+        151,
+        152,
+        153,
+        199,
+        307,
+        377,
+        522,
+        37,
+        38,
+        41,
+        42,
+        46,
+        47,
+        48,
+        49,
+        50,
+        51,
+        52,
+        55,
+        56,
+        58,
+        106,
+        145,
+        146,
+        147,
+        148,
+        149,
+        197,
+        198,
+        308,
+        325,
+        495,
+        271,
+        523,
+    ]
+    actual_ids.sort()
+    assert ids == actual_ids

tests/test_helper.py

@@ -0,0 +1,19 @@
+"""Test helper module"""
+
+from BCSFE_Python import helper
+
+
+def test_str_to_gv():
+    """Test that a game version with . is converted to an int representation"""
+    assert helper.str_to_gv("1.0.0") == "10000"
+    assert helper.str_to_gv("11.8.2") == "110802"
+    assert helper.str_to_gv("1.7") == "10700"
+    assert helper.str_to_gv("10.87") == "108700"
+
+
+def test_gv_to_str():
+    """Test that an int representation is converted to a game version with ."""
+    assert helper.gv_to_str(10000) == "1.0.0"
+    assert helper.gv_to_str(110802) == "11.8.2"
+    assert helper.gv_to_str(10700) == "1.7.0"
+    assert helper.gv_to_str(108700) == "10.87.0"