def test_count(self): with HTTMock(create_mock()): relation = MyriaRelation(FULL_NAME, connection=self.connection) count = relation.count() self.assertListEqual(count.query.grouping_list, []) self.assertListEqual(count.query.aggregate_list, [COUNTALL()]) self.assertIsNotNone(count._sink().to_json())
def test_count_attribute(self): with HTTMock(create_mock()): relation = MyriaRelation(FULL_NAME, connection=self.connection) count = relation.count('column2') self.assertListEqual(count.query.grouping_list, []) self.assertListEqual(count.query.aggregate_list, [COUNT(UnnamedAttributeRef(1))]) self.assertIsNotNone(count._sink().to_json())