Skip to content

Commit

Permalink
Merge 6932975 into 72136fc
Browse files Browse the repository at this point in the history
  • Loading branch information
anaschwendler committed Dec 1, 2017
2 parents 72136fc + 6932975 commit 1bfcc5a
Show file tree
Hide file tree
Showing 2 changed files with 10 additions and 10 deletions.
2 changes: 1 addition & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -34,5 +34,5 @@
'serenata_toolbox.datasets'
],
url=REPO_URL,
version='12.3.1'
version='12.4.1'
)
18 changes: 9 additions & 9 deletions tests/journey/test_chamber_of_deputies_deputies_dataset.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,22 +11,22 @@ def setUp(self):

def test_fetch(self):
df = self.subject.fetch()
actualColumns = df.columns
actualColumns = set(df.columns)

expectedColumns = [
expectedColumns = {
'congressperson_id', 'budget_id', 'condition',
'congressperson_document', 'civil_name', 'congressperson_name',
'picture_url', 'gender', 'state', 'party', 'phone_number', 'email'
]
}
self.assertTrue((np.array(expectedColumns) == np.array(actualColumns)).all())

expectedGenders = ['male', 'female']
actualGenders = df.gender.unique()
self.assertTrue((np.array(expectedGenders) == np.array(actualGenders)).all())
expectedGenders = {'male', 'female'}
actualGenders = set(df.gender.unique())
self.assertTrue(expectedGenders, actualGenders)

expectedConditions = ['Holder', 'Substitute']
actualConditions = df.condition.unique()
self.assertTrue((np.array(expectedConditions) == np.array(actualConditions)).all())
expectedConditions = {'Substitute', 'Holder'}
actualConditions = set(df.condition.unique())
self.assertEqual(expectedConditions, actualConditions)

if __name__ == '__main__':
main()

0 comments on commit 1bfcc5a

Please sign in to comment.