initial commit

tests/test_ai.py

@@ -0,0 +1,77 @@
+import json
+
+import pytest
+
+import ai
+
+
+@pytest.mark.skip
+def test_classify_live():
+    item_names = [
+        'Syston Cable Technology Cat 6A+ Ethernet Network Cable，700MHz 23AWG Solid Bare Copper Wire Outdoor/Indoor, No Ends 100/250/500/1000 Ft Available, Heat Resistant Plenum Rated (100 FT, CMP, Blue-CMP)',
+        "Chef'n Salad Dressing Mixer, Baking White/Vintage Indigo",
+        "Lee Women's Ultra Lux Comfort with Flex Motion Skinny Leg Jean with Ever Fit, Black",
+    ]
+
+    categories = [
+        'Home: Mortgage',
+        'Home: Property Tax',
+        'Home: Home Insurance',
+        'Home: Water',
+        'Home: Oil',
+        'Home: Other Utilities',
+        'Car: Car Insurance - Annual',
+        'Car: Car Insurance - Monthly',
+        'Car: Maintenance',
+        'Internal Master Category: Inflow: Ready to Assign',
+        'Internal Master Category: Deferred Income SubCategory',
+        'Internal Master Category: Uncategorized',
+        'Essentials: Groceries',
+        "Essentials: I'm Too Lazy To Cook",
+        'Essentials: Gas',
+        'Essentials: Clothes',
+        'Essentials: Baby',
+        'Essentials: Toiletries and Hygiene',
+        'Essentials: Medical',
+        'Essentials: Software Subscriptions',
+        'Essentials: Stuff I Forgot to Budget For',
+        'Essentials: Charity',
+        'Semi-Essentials: Sports and Fitness',
+        'Semi-Essentials: Gifts',
+        'Semi-Essentials: Technology',
+        'Semi-Essentials: Household Items',
+        'Semi-Essentials: Travel',
+        'Credit Card Payments: Citi Credit',
+        'Credit Card Payments: BoA Credit',
+        'Credit Card Payments: Amazon Credit',
+        "Credit Card Payments: BJ's Credit",
+        'Debt Payments: Student Loan',
+        'Just for Fun: Date Night',
+        'Just for Fun: Hobbies/Activities',
+        'Just for Fun: Doing Things With Friends',
+        'Just for Fun: Gaming',
+        'Just for Fun: Media',
+        'Long-term: Home Improvement',
+        'Long-term: The Future'
+    ]
+
+    res = ai.classify(item_names, categories)
+    assert len(res) == 3
+    expected_categories = [
+        'Semi-Essentials: Technology',
+        'Semi-Essentials: Household Items',
+        'Essentials: Clothes'
+    ]
+    for classified, ec in zip(res.values(), expected_categories):
+        assert classified.category == ec
+
+
+@pytest.mark.skip
+def test_classify_live_trouble():
+    with open('output/item_names.json') as f:
+        item_names = json.load(f)
+    with open('output/text_categories.json') as f:
+        text_categories = json.load(f)
+
+    res = ai.classify(item_names, text_categories)
+    assert set(item_names) ^ res.keys() == set()