Skip to content

Commit

Permalink
added tests for remote.client scripts
Browse files Browse the repository at this point in the history
  • Loading branch information
ZueFe committed Jul 8, 2018
1 parent a00d125 commit 69faf35
Showing 1 changed file with 61 additions and 0 deletions.
61 changes: 61 additions & 0 deletions test/remote/test_download_stats.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,61 @@
import os
import sys

sys.path.insert(0, os.path.abspath('.'))

import unittest
import numpy.testing as npt
import remote.client.download_stats as dw

class TestDownload(unittest.TestCase):
def test_individual_batters(self):
df = dw.load_individual_batters()

assert 'bat' in df.file_name
assert 'individual' in df.file_name

def test_individual_field(self):
df = dw.load_individual_field()

assert 'field' in df.file_name
assert 'individual' in df.file_name

def test_individual_pitchers(self):
df = dw.load_individual_pitchers()

assert 'pitcher' in df.file_name
assert 'individual' in df.file_name

def test_individual_catchers(self):
df = dw.load_individual_catchers()

assert 'catcher' in df.file_name
assert 'individual' in df.file_name

def test_team_batters(self):
df = dw.load_team_batters()

assert 'bat' in df.file_name
assert 'team' in df.file_name

def test_team_field(self):
df = dw.load_team_fields()

assert 'field' in df.file_name
assert 'team' in df.file_name

def test_team_pitchers(self):
df = dw.load_team_pitchers()

assert 'pitcher' in df.file_name
assert 'team' in df.file_name

def test_team_catcher(self):
df = dw.load_team_catchers()

assert 'catcher' in df.file_name
assert 'team' in df.file_name

def test_error(self):
with npt.assert_raises(Exception):
df = dw.load_stats({'category': 'not', 'type': '4'})

0 comments on commit 69faf35

Please sign in to comment.